Update app.py

app.py

@@ -41,7 +41,7 @@ def compute_text_embeddings(list_of_strings):
     
 def download_img(path):
     img_data = requests.get(path).content
-    local_path = path.split("/")[-1] + ".jpg"
+    local_path = path.split("/")[-1]
     with open(local_path, 'wb') as handler:
         handler.write(img_data)
     return local_path
@@ -58,9 +58,10 @@ def predict(im):
     n_results=3
     text_embeddings = compute_text_embeddings([query]).detach().numpy()
     results = np.argsort((embeddings@text_embeddings.T)[:, 0])[-1:-n_results-1:-1]
-    paths = [download_img(df.iloc[i]['path']) for i in results]
-    print(paths)
-    return {LABELS[i]: v.item() for i, v in zip(indices, values)}, paths
+    outputs = [download_img(df.iloc[i]['path']) for i in results]
+    outputs.insert(0, {LABELS[i]: v.item() for i, v in zip(indices, values)})
+    print(outputs)
+    return outputs
 
 title = "Draw to Search"
 iface = gr.Interface(