Update app.py

app.py

@@ -16,7 +16,7 @@ model = FlamingoModel.from_pretrained('dhansmair/flamingo-mini')
 model.to(device)
 model.eval()
 
-processor = FlamingoProcessor(model.config, load_vision_processor=True)
+processor = FlamingoProcessor(model.config, device=device)
 
 # setup some example images
 examples = []
@@ -29,10 +29,11 @@ if os.path.isdir(EXAMPLES_DIR):
 def predict_caption(image, prompt):
     assert isinstance(prompt, str)
     
-    features = processor.extract_features(image).to(device)
-    caption = model.generate_captions(processor, 
-            visual_features=features,
-            prompt=prompt)
+    caption = model.generate_captions(
+        processor, 
+        images=image,
+        prompt=prompt
+    )
   
     if isinstance(caption, list):
         caption = caption[0]