Update app.py

app.py

@@ -30,11 +30,8 @@ def calculate_score(image, text, model_name):
     model = models[model_name]
     processor = processors[model_name]
     
-    # Get the correct image size for the model
-    _, image_size = CLIP_MODELS[model_name]
-    
     # Preprocess the image and text
-    inputs = processor(text=labels, images=image, return_tensors="pt", padding=True)
+    inputs = processor(text=labels, images=[image], return_tensors="pt", padding=True)
     inputs = {k: v.to("cuda") for k, v in inputs.items()}
     
     # Calculate scores
@@ -43,7 +40,7 @@ def calculate_score(image, text, model_name):
     
     logits_per_image = outputs.logits_per_image.cpu().numpy()
     
-    results_dict = {label: score / 100.0 for label, score in zip(labels, logits_per_image[0])}
+    results_dict = {label: float(score) for label, score in zip(labels, logits_per_image[0])}
     return results_dict
 
 with gr.Blocks() as demo:
@@ -63,23 +60,12 @@ with gr.Blocks() as demo:
             return None
         return calculate_score(image, text, model_name)
     
-    image_input.change(
-        fn=process_inputs,
-        inputs=[image_input, text_input, model_dropdown],
-        outputs=output_label
-    )
-    
-    text_input.submit(
-        fn=process_inputs,
-        inputs=[image_input, text_input, model_dropdown],
-        outputs=output_label
-    )
+    inputs = [image_input, text_input, model_dropdown]
+    outputs = output_label
     
-    model_dropdown.change(
-        fn=process_inputs,
-        inputs=[image_input, text_input, model_dropdown],
-        outputs=output_label
-    )
+    image_input.change(fn=process_inputs, inputs=inputs, outputs=outputs)
+    text_input.submit(fn=process_inputs, inputs=inputs, outputs=outputs)
+    model_dropdown.change(fn=process_inputs, inputs=inputs, outputs=outputs)
     
     gr.Examples(
         examples=[
@@ -90,8 +76,8 @@ with gr.Blocks() as demo:
             ]
         ],
         fn=process_inputs,
-        inputs=[image_input, text_input, model_dropdown],
-        outputs=output_label,
+        inputs=inputs,
+        outputs=outputs,
     )
 
 demo.launch()