Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -80,7 +80,7 @@ def generate_output(input_text):
|
|
| 80 |
# Generate outputs
|
| 81 |
outputs = model.generate(**inputs, max_new_tokens=2500, use_cache=True)
|
| 82 |
output = tokenizer.batch_decode(outputs)
|
| 83 |
-
cleaned_response = process_output(output)
|
| 84 |
return output,cleaned_response
|
| 85 |
|
| 86 |
|
|
|
|
| 80 |
# Generate outputs
|
| 81 |
outputs = model.generate(**inputs, max_new_tokens=2500, use_cache=True)
|
| 82 |
output = tokenizer.batch_decode(outputs)
|
| 83 |
+
cleaned_response = process_output(str(output))
|
| 84 |
return output,cleaned_response
|
| 85 |
|
| 86 |
|