Update app.py

app.py

@@ -31,7 +31,7 @@ def run(image: Union[str, PIL.Image.Image]) -> str:
     if isinstance(image, str):
         image = Image.open(image)
     inputs = processor(images=image, return_tensors="pt").to(device)
-    out = model.generate(pixel_values=inputs.pixel_values, num_beams=3, max_length=300)
+    out = model.generate(pixel_values=inputs.pixel_values, num_beams=3, repitition_penalty=2.5, max_length=300)
     generated_caption = processor.decode(out[0], skip_special_tokens=True)
     return generated_caption