Spaces:
Sleeping
Sleeping
outputs[1]
Browse files
app.py
CHANGED
|
@@ -94,7 +94,7 @@ def predict_text(text, url = 'https://arinsight.co/2024_FA_AEC_1200_GR1_GR2.pdf'
|
|
| 94 |
outputs = model.generate(**inputs, max_new_tokens=1024)
|
| 95 |
|
| 96 |
# Decode the output to return the final response
|
| 97 |
-
response = processor.decode(outputs[
|
| 98 |
return response
|
| 99 |
|
| 100 |
|
|
|
|
| 94 |
outputs = model.generate(**inputs, max_new_tokens=1024)
|
| 95 |
|
| 96 |
# Decode the output to return the final response
|
| 97 |
+
response = processor.decode(outputs[1], skip_special_tokens=True, skip_prompt=True)
|
| 98 |
return response
|
| 99 |
|
| 100 |
|