Update app.py
Browse files
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,
|