Update app.py

app.py

@@ -21,7 +21,7 @@ import glob
 
 # Model
 
-def predict(text,img):
+def predict(img,text):
   model = torch.hub.load('ultralytics/yolov5', 'yolov5s')
   from PIL import Image
   img1 = Image.open(img).convert("RGB")
@@ -75,7 +75,7 @@ def predict(text,img):
 
   with torch.no_grad():
       # Encode and normalize the description using CLIP
-      text_encoded = model.encode_text(clip.tokenize([text]).to(device))
+      text_encoded = model.encode_text(clip.tokenize(text).to(device))
       text_encoded /= text_encoded.norm(dim=-1, keepdim=True)
 
   similarity = text_encoded.cpu().numpy() @ image_features.cpu().numpy().T
@@ -85,11 +85,11 @@ def predict(text,img):
   #ipyplot.plot_images(imgs,img_width=350)
   return imgs[0]
 
-text = gr.inputs.Textbox(lines=1, label="Text query", placeholder="Introduce the search text...",)
+#text = gr.inputs.Textbox(lines=1, label="Text query", placeholder="Introduce the search text...",)
 #img = gr.inputs.Image()
 
 #img = "image"
 
 
 
-gr.Interface(predict, ["image", text], outputs="image", title='Search inside image').launch();
+gr.Interface(predict, ["image", gr.inputs.Textbox(lines=1, label="Text query", placeholder="Introduce the search text...",)], outputs="image", title='Search inside image').launch();