Update app.py

app.py

@@ -7,15 +7,15 @@ import torch
 # Load JinaAI CLIP model
 model = AutoModel.from_pretrained("jinaai/jina-clip-v1", trust_remote_code=True)
 
-# Function to process input (convert text or image)
+# Function to process input
 def process_input(input_data, input_type):
     if input_type == "Text":
         return model.encode_text([input_data]) if input_data.strip() else None
     elif input_type == "Image":
-        if isinstance(input_data, np.ndarray):  # Gradio provides image as NumPy array
-            image = Image.fromarray(input_data)  # Convert NumPy array to PIL Image
-            return model.encode_image([image])
-        return None  # Invalid input type
+        if isinstance(input_data, np.ndarray):  # Gradio provides NumPy array for images
+            image = Image.fromarray(input_data)  # Convert NumPy to PIL Image
+            return model.encode_image(image)  # Directly pass image (no list)
+        return None  # If input is not valid
     return None
 
 # Function to compute similarity
@@ -41,12 +41,12 @@ def compute_similarity(input1, input2, input1_type, input2_type):
     similarity_score = (embedding1 @ embedding2.T).item()
     return f"Similarity Score: {similarity_score:.4f}"
 
-# Function to toggle input fields dynamically and clear previous values
+# Function to update UI dynamically
 def update_visibility(input1_type, input2_type):
     return (
-        gr.update(visible=(input1_type == "Text"), value="" if input1_type == "Image" else None),  
-        gr.update(visible=(input1_type == "Image"), value=None), 
-        gr.update(visible=(input2_type == "Text"), value="" if input2_type == "Image" else None),  
+        gr.update(visible=(input1_type == "Text"), value="" if input1_type == "Image" else None),
+        gr.update(visible=(input1_type == "Image"), value=None),
+        gr.update(visible=(input2_type == "Text"), value="" if input2_type == "Image" else None),
         gr.update(visible=(input2_type == "Image"), value=None)
     )
 
@@ -68,7 +68,7 @@ with gr.Blocks() as demo:
 
     output = gr.Textbox(label="Similarity Score / Error", interactive=False)
 
-    # Toggle visibility of inputs dynamically and clear values
+    # Toggle visibility of inputs dynamically
     input1_type.change(update_visibility, inputs=[input1_type, input2_type], 
                        outputs=[input1_text, input1_image, input2_text, input2_image])
     input2_type.change(update_visibility, inputs=[input1_type, input2_type],