Upload app.py

app.py

@@ -17,14 +17,15 @@ model = SentenceTransformer('sentence-transformers/all-mpnet-base-v2')
 
 def generate_memes(prompt):
     prompt_embedding = model.encode(prompt, convert_to_tensor=True)
-    hits = util.semantic_search(prompt_embedding, embeddings, top_k=5)
+    hits = util.semantic_search(prompt_embedding, embeddings, top_k=6)
     hits = pd.DataFrame(hits[0], columns=['corpus_id', 'score'])
     desired_ids = hits["corpus_id"]
     filtered_df = df.loc[df['id'].isin(desired_ids)]
     filtered_list = list(filtered_df["url"]) 
-    images = [Image.open(BytesIO(requests.get(img).content)) for img in filtered_list]
+    # images = [Image.open(BytesIO(requests.get(img).content)) for img in filtered_list]
     return (
-       images
+    #    images
+    filtered_list
     )
 input_textbox = gr.inputs.Textbox(lines=1, label="Search something cool")
 output_gallery = gr.Gallery(