Update app.py

app.py

@@ -63,8 +63,8 @@ def caption_random_image():
     caption_eng = processor.decode(out[0], skip_special_tokens=True)
 
     # Translate caption from English to Spanish
-    trans_inputs = trans_tokenizer.encode(caption_en, return_tensors="pt")
-    trans_out = trans_model.generate(**tokenizer(trans_inputs, return_tensors="pt", padding=True))
+    trans_inputs = trans_tokenizer.encode(caption_eng, return_tensors="pt")
+    trans_out = trans_model.generate(trans_inputs)
     caption_es = trans_tokenizer.decode(trans_out[0], skip_special_tokens=True)
 
     return image, caption_eng, caption_es