Update app.py

app.py

@@ -42,8 +42,8 @@ def compare_image_to_text_models(image, prompt, model1_choice, model1_custom, mo
 
     # Create pipelines for image-to-text.
     # These models should support a call signature of (image, prompt)
-    pipe1 = pipeline("image-to-text", model=model1_name, device=device)
-    pipe2 = pipeline("image-to-text", model=model2_name, device=device)
+    pipe1 = pipeline(task="image-text-to-text", model=model1_name, device=device)
+    pipe2 = pipeline(task="image-text-to-text", model=model2_name, device=device)
 
     # Run inference on the image with the provided prompt.
     output1 = pipe1(image, query=prompt)