Upload app.py

app.py

@@ -14,9 +14,12 @@ def prepare_image(image_path):
 def generate_caption(pixel_values):
     model.eval()
     with torch.no_grad():
-        generated_ids = model.generate(
-            pixel_values=pixel_values,
-            max_length=500)
+            generated_ids = model.generate(pixel_values=pixel_values, max_length=200,
+            do_sample=True,
+            top_k=120,
+            top_p=0.9,
+            early_stopping=True,
+            num_return_sequences=1)
     return processor.batch_decode(generated_ids, skip_special_tokens=True)[0]
 
 def translate_to_portuguese(text):