Update caption.py

caption.py

@@ -15,7 +15,7 @@ use_fp16 = device.type == "cuda"
 if use_fp16:
     model.half()
 
-max_length = 16
+max_length = 50
 num_beams = 4
 gen_kwargs = {'max_length': max_length, 'num_beams': num_beams}