Update app.py

app.py

@@ -12,7 +12,7 @@ pipe = None
 sample = None
 model_repo = 'Solenya-ai/dill'
 
-def search(image=None, query=None, combination=1, selection=1):
+def search(image=None, query=None):
     global pipe
     global sample
 
@@ -30,7 +30,7 @@ def search(image=None, query=None, combination=1, selection=1):
         image_embed = np.array(pipe.encode(image))
 
     if query is not None and image is not None:
-        embed = text_embed*(image_embed**selection) + combination*image_embed
+        embed = text_embed*(image_embed**0.2)
     elif image is not None:
         embed = image_embed
     else:
@@ -42,8 +42,7 @@ def search(image=None, query=None, combination=1, selection=1):
 
 iface = gr.Interface(search,
                         [gr.inputs.Image(type="pil"), 
-                         gr.inputs.Textbox(lines=1, placeholder="Type here to search for similar images..."), 
-                         gr.inputs.Slider(-1, 2, default=0.2, step=0.1), gr.inputs.Slider(-1, 2, default=0, step=0.1)],
+                         gr.inputs.Textbox(lines=1, placeholder="Type here to search for similar images...")],
                         gr.Gallery(columns=3),
                         title="Search for similar images",
 )