Update run.py

run.py

@@ -21,7 +21,7 @@ def check(x):
 def process_image(image):
     pixel_values = processor(image, return_tensors="pt").pixel_values
     with torch.no_grad():
-        generated_ids = model.generate(pixel_values, num_beams=4, num_return_sequences=4)
+        generated_ids = model.generate(pixel_values, num_beams=2, num_return_sequences=2)
     generated_text = processor.batch_decode(generated_ids, skip_special_tokens=True)
     generated_text = [x for x in generated_text if check(x)]
     return generated_text[0]