Update app.py
Browse files
app.py
CHANGED
|
@@ -84,7 +84,7 @@ def code_summary(inputs,sum_prompt,num_beam, sec):
|
|
| 84 |
generated_ids = sum_model.generate(input_ids, max_length=25,do_sample=False,num_beams=num_beam)
|
| 85 |
output = tokenizer.decode(generated_ids[0], skip_special_tokens=True)
|
| 86 |
elif sec == "Java":
|
| 87 |
-
input_ids = tokenizer.encode(inputs + sum_prompt, return_tensors='pt'
|
| 88 |
generated_ids = sum_model.generate(input_ids, max_length=25,do_sample=False,num_beams=num_beam)
|
| 89 |
output = tokenizer.decode(generated_ids[0], skip_special_tokens=True)
|
| 90 |
s3.pop(output)
|
|
|
|
| 84 |
generated_ids = sum_model.generate(input_ids, max_length=25,do_sample=False,num_beams=num_beam)
|
| 85 |
output = tokenizer.decode(generated_ids[0], skip_special_tokens=True)
|
| 86 |
elif sec == "Java":
|
| 87 |
+
input_ids = tokenizer.encode(inputs + sum_prompt, return_tensors='pt')
|
| 88 |
generated_ids = sum_model.generate(input_ids, max_length=25,do_sample=False,num_beams=num_beam)
|
| 89 |
output = tokenizer.decode(generated_ids[0], skip_special_tokens=True)
|
| 90 |
s3.pop(output)
|