app.py

app.py

@@ -112,13 +112,14 @@ def image_search(search_text, search_image, option):
       #matched_images = find_matches(text_features, photo_features, photo_ids, 4)
       #similarity = util.cos_sim(image_emb, img_emb)
       #matched_results = util.semantic_search(image_emb, img_emb, 4)[0]
+      
       image_emb = model.encode([Image.open(search_image)], convert_to_tensor=True)
       similarity = util.cos_sim(img_emb, image_emb)
       return display_matches(similarity)
   
 gr.Interface(fn=image_search, 
             inputs=[gr.inputs.Textbox(lines=7, label="Input Text"),
-                    gr.inputs.Image(type="pil", optional=True),
+                    gr.inputs.Image(optional=True),
                     gr.inputs.Dropdown(["Text-To-Image", "Image-To-Image"])
                     ], 
             outputs=gr.outputs.Carousel([gr.outputs.Image(type="pil"), gr.outputs.Image(type="pil"), gr.outputs.Image(type="pil"), gr.outputs.Image(type="pil")]),