Upload app.py

app.py

@@ -3,8 +3,9 @@ from huggingface_hub import hf_hub_download
 import os
 import pickle
 import pandas as pd
-import urllib.request
 from PIL import Image
+import requests
+from io import BytesIO
 import gradio as gr
 
 pd.options.mode.chained_assignment = None  # Turn off SettingWithCopyWarning
@@ -21,7 +22,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 = [Image.open(urllib.request.urlretrieve(img)) for img in filtered_list]
+    images = [Image.open(BytesIO(requests.get(img).content)) for img in filtered_list]
     return (
        images
     )