Upload app.py

app.py

@@ -3,6 +3,8 @@ from huggingface_hub import hf_hub_download
 import os
 import pickle
 import pandas as pd
+import urllib.request
+from PIL import Image
 import gradio as gr
 
 pd.options.mode.chained_assignment = None  # Turn off SettingWithCopyWarning
@@ -19,7 +21,7 @@ def generate_memes(prompt):
     desired_ids = hits["corpus_id"]
     filtered_df = df.loc[df['id'].isin(desired_ids)]
     filtered_list = list(filtered_df["url"]) 
-    images = [gr.Image.update(value=img, visible=True) for img in filtered_list]
+    images = [Image.open(urllib.request.urlretrieve(img)) for img in filtered_list]
     return (
        images
     )