NickNYU commited on
Commit
12a7db6
·
1 Parent(s): 30c0ce1

print all results

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -24,7 +24,7 @@ def predict(image):
24
 
25
  preds = tokenizer.batch_decode(output_ids, skip_special_tokens=True)
26
  preds = [pred.strip() for pred in preds]
27
- return preds[0]
28
 
29
  examples_folder = os.path.join(os.path.dirname(__file__), "examples")
30
  examples = [os.path.join(examples_folder, file) for file in os.listdir(examples_folder)]
 
24
 
25
  preds = tokenizer.batch_decode(output_ids, skip_special_tokens=True)
26
  preds = [pred.strip() for pred in preds]
27
+ return ", ".join(preds)
28
 
29
  examples_folder = os.path.join(os.path.dirname(__file__), "examples")
30
  examples = [os.path.join(examples_folder, file) for file in os.listdir(examples_folder)]