Antoine Chaffin commited on
Commit
30c1e0f
·
1 Parent(s): 3f46c94

Remove double truncation of the output text

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -77,7 +77,7 @@ def generate_answer(query, retrieved_image):
77
  skip_special_tokens=True,
78
  clean_up_tokenization_spaces=False,
79
  )
80
- return output_text[0]
81
 
82
 
83
  def query_index(query, index):
 
77
  skip_special_tokens=True,
78
  clean_up_tokenization_spaces=False,
79
  )
80
+ return output_text
81
 
82
 
83
  def query_index(query, index):