image-to-music

Runtime error

ybelkada commited on Dec 17, 2022

Commit

5cad334

1 Parent(s): 1a6a022

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -18,7 +18,7 @@ blip_model = BlipForConditionalGeneration.from_pretrained(blip_model_id)
 processor = BlipProcessor.from_pretrained(blip_model_id)
 def predict(image):
-    inputs = processor(image)
     output_blip = blip_model.generate(**inputs)
     prompt = processor.decode(output_blip[0], skip_special_tokens=True)

 processor = BlipProcessor.from_pretrained(blip_model_id)
 def predict(image):
+    inputs = processor(image, return_tensors="pt")
     output_blip = blip_model.generate(**inputs)
     prompt = processor.decode(output_blip[0], skip_special_tokens=True)