Update caption.py

caption.py

@@ -16,7 +16,7 @@ if use_fp16:
     model.half()
 
 # Set generation parameters
-gen_kwargs = {'max_length': 92, 'num_beams': 1}  # Adjust num_beams if needed
+gen_kwargs = {'max_length': 90, 'num_beams': 2}  # Adjust num_beams if needed
 
 
 # Prediction function