Update app.py

app.py

@@ -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]