nsy commited on
Commit
77b0324
·
1 Parent(s): 8ca3f79

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -179,7 +179,7 @@ def code_generation(inputs,gene_prompt,top_p, temperature,sec):
179
  top_p=top_p)
180
  output = tokenizer.decode(generated_ids[0], skip_special_tokens=True)
181
  elif sec == "Java":
182
- input_ids = tokenizer.encode(inputs + gene_prompt, return_tensors='pt'
183
  generated_ids = gene_model.generate(
184
  input_ids,
185
  do_sample=True,
 
179
  top_p=top_p)
180
  output = tokenizer.decode(generated_ids[0], skip_special_tokens=True)
181
  elif sec == "Java":
182
+ input_ids = tokenizer.encode(inputs + gene_prompt, return_tensors='pt')
183
  generated_ids = gene_model.generate(
184
  input_ids,
185
  do_sample=True,