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 |
|