SeyedAli commited on
Commit
b10bf37
1 Parent(s): df36445

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -31,7 +31,7 @@ def predict_step(image_paths):
31
 
32
  preds = tokenizer.batch_decode(output_ids, skip_special_tokens=True)
33
  preds = [pred.strip() for pred in preds]
34
- return run_transaltion_model(preds[0])[0]
35
 
36
  def ImageCaptioning(image):
37
  with tempfile.NamedTemporaryFile(suffix=".png") as temp_image_file:
 
31
 
32
  preds = tokenizer.batch_decode(output_ids, skip_special_tokens=True)
33
  preds = [pred.strip() for pred in preds]
34
+ return preds[0]
35
 
36
  def ImageCaptioning(image):
37
  with tempfile.NamedTemporaryFile(suffix=".png") as temp_image_file: