bug fix for new model

app.py

@@ -40,20 +40,19 @@ def find_similar_images(query_embedding, image_embeddings, k=2):
     return closest_indices, scores
 
 
-def main(query, model_id="rclip", k=2):
-    if model_id=="rclip":
+def main(query, model_id="RCLIP", k=2):
+    if model_id=="RCLIP":
         # Load RCLIP model
         model = VisionTextDualEncoderModel.from_pretrained(model_path_rclip)
         processor = VisionTextDualEncoderProcessor.from_pretrained(model_path_rclip)
         # Load image embeddings 
         image_embeddings = load_embeddings(embeddings_file_rclip)
-    elif mode_id=="pubmedclip":
+    elif model_id=="PubMedCLIP":
         model = CLIPModel.from_pretrained(model_path_pubmedclip)
         processor = CLIPProcessor.from_pretrained(model_path_pubmedclip)
         # Load image embeddings 
         image_embeddings = load_embeddings(embeddings_file_pubmedclip)   
 
-
     # Embed the query
     inputs = processor(text=query, images=None, return_tensors="pt", padding=True)
     with torch.no_grad():
@@ -75,10 +74,12 @@ def main(query, model_id="rclip", k=2):
 
 # Define the Gradio interface
 examples = [
-            ["Chest X-ray photos",5],
-            ["Orthopantogram (OPG)",5],
-            ["Brain Scan",5], 
-            ["tomography",5]
+            ["Chest X-ray photos", "RCLIP", 5],
+            ["Chest X-ray photos", "PubMedCLIP", 5],
+            ["Orthopantogram (OPG)", "RCLIP",5],
+            ["Brain Scan", "RCLIP",5], 
+            ["Chest X-ray photos", "PubMedCLIP", 5],
+            ["tomography", "RCLIP",5],
 ]
 
 title="RCLIP Image Retrieval"
@@ -91,24 +92,25 @@ with gr.Blocks(title=title) as demo:
             gr.Markdown(description)
         gr.HTML(value="<img src=\"https://newresults.co.uk/wp-content/uploads/2022/02/teesside-university-logo.png\" alt=\"teesside logo\" width=\"120\" height=\"70\">", show_label=False,scale=1)
         #Image.open("./data/teesside university logo.png"), height=70, show_label=False, container=False)
-    with gr.Column(variant="compact"):
-        with gr.Row(variant="compact"):
-            query = gr.Textbox(value="Chest X-Ray Photos", label="Enter your query", show_label=False, placeholder= "Enter your query" , scale=5)
-            btn = gr.Button("Search query", variant="primary", scale=1)
-        
-        n_s = gr.Slider(2, 10, label='Number of Top Results', value=5, step=1.0, show_label=True)
+    with gr.Row(variant="compact"):
+         query = gr.Textbox(value="Chest X-Ray Photos", label="Enter your query", show_label=False, placeholder= "Enter your query" , scale=5)
+         btn = gr.Button("Search query", variant="primary", scale=1)
+         
+    with gr.Row(variant="compact"):   
+        model_id = gr.Dropdown(["RCLIP", "PubMedCLIP"], value="RCLIP", label="Model", type="value", scale=1)
+        n_s = gr.Slider(2, 10, label='Number of Top Results', value=5, step=1.0, show_label=True, scale=1)
         
-    
+
     with gr.Column(variant="compact"):
         gr.Markdown("## Results")
         gallery = gr.Gallery(label="found images", show_label=True, elem_id="gallery", columns=[2], rows=[4], object_fit="contain", height="400px", preview=True)
         gr.Markdown("Information of the found images")
         df = gr.DataFrame()
-    btn.click(main, [query, n_s], [gallery, df])
+    btn.click(main, [query, model_id, n_s], [gallery, df])
     
     with gr.Column(variant="compact"):
         gr.Markdown("## Examples")
-        gr.Examples(examples, [query, n_s])
+        gr.Examples(examples, [query, model_id, n_s])
     
     
 demo.launch(debug='True')