Update app.py

app.py

@@ -46,8 +46,8 @@ def compare_image_to_text_models(image, prompt, model1_choice, model1_custom, mo
     pipe2 = pipeline("image-to-text", model=model2_name, device=device)
 
     # Run inference on the image with the provided prompt.
-    output1 = pipe1(image, prompt)
-    output2 = pipe2(image, prompt)
+    output1 = pipe1(image, query=prompt)
+    output2 = pipe2(image, query=prompt)
 
     # Extract the generated text.
     def extract_text(output):