khurrameycon commited on
Commit
7f7c55c
·
verified ·
1 Parent(s): 96e459a

outputs[1]

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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[0], skip_special_tokens=True, skip_prompt=True)
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