Update caption.py

caption.py

@@ -15,8 +15,9 @@ use_fp16 = device.type == "cuda"
 if use_fp16:
     model.half()
 
-# Set generation parameters
-gen_kwargs = {'max_length': 50, 'num_beams': 2}  # Adjust num_beams if needed
+max_length = 16
+num_beams = 4
+gen_kwargs = {'max_length': max_length, 'num_beams': num_beams}
 
 
 # Prediction function