comparing-captioning-models

Runtime error

nielsr HF staff commited on Jan 2, 2023

Commit

38ab1e3

1 Parent(s): 075ee8a

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -16,7 +16,7 @@ git_model.to(device)
 blip_model.to(device)
 def generate_caption(processor, model, image):
-    inputs = processor(image=image, return_tensors="pt").to(device)
     generated_ids = model.generate(pixel_values=inputs.pixel_values, max_length=50)

 blip_model.to(device)
 def generate_caption(processor, model, image):
+    inputs = processor(images=image, return_tensors="pt").to(device)
     generated_ids = model.generate(pixel_values=inputs.pixel_values, max_length=50)