Spaces:

Nepjune
/

Project_image_captioning_blip

Runtime error

Nepjune commited on Feb 6, 2024

Commit

01185b8

verified ·

1 Parent(s): 5199324

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -11,7 +11,7 @@ model.to(device)
 max_length = 16
 num_beams = 4
-gen_kwargs = ("max_length" : max_length, "num_beams" : num_beams)
 def predict_caption(image_paths):
     images = []

 max_length = 16
 num_beams = 4
+gen_kwargs = {"max_length": max_length, "num_beams": num_beams}
 def predict_caption(image_paths):
     images = []