Spaces:
Runtime error
Runtime error
Corrected errors
Browse files
app.py
CHANGED
|
@@ -33,7 +33,7 @@ def generate_answer(model_name,question):
|
|
| 33 |
outputs = model.generate(**inputs, max_new_tokens = 64, use_cache = True)
|
| 34 |
decoded_output = tokenizer.batch_decode(outputs,skip_special_tokens=True)[0]
|
| 35 |
predicted_title = nltk.sent_tokenize(decoded_output.strip())[0]
|
| 36 |
-
|
| 37 |
|
| 38 |
|
| 39 |
iface = gr.Interface(
|
|
|
|
| 33 |
outputs = model.generate(**inputs, max_new_tokens = 64, use_cache = True)
|
| 34 |
decoded_output = tokenizer.batch_decode(outputs,skip_special_tokens=True)[0]
|
| 35 |
predicted_title = nltk.sent_tokenize(decoded_output.strip())[0]
|
| 36 |
+
return predicted_title
|
| 37 |
|
| 38 |
|
| 39 |
iface = gr.Interface(
|