Update app.py

app.py

@@ -14,6 +14,11 @@ import time
 import replicate
 import torch
 import pickle
+from urllib.request import urlopen
+
+from PIL import Image
+import requests
+from io import BytesIO
 
 prompt_list = [p for p in list(set(
                 pd.read_csv('./twitter_prompts.csv').iloc[:, 1].tolist())) if type(p) == str]
@@ -56,8 +61,10 @@ def next_image():
             "rynmurdock/zahir:0bf6ef3012f23397a6f31d4e4066ff9a13f028e54b119041cb82b0b38fdc6a36",
             input={"prompt": prompt,}
             )
+            response = requests.get(url)
+            image = Image.open(BytesIO(response.content))
 
-            embs.append(pooled_embeds)
+            embs.append(pickle.load(urlopen(pooled_embeim_embds, 'rb')))
             return image
         else:
             print('######### Roaming #########')
@@ -94,7 +101,11 @@ def next_image():
             "rynmurdock/zahir:0bf6ef3012f23397a6f31d4e4066ff9a13f028e54b119041cb82b0b38fdc6a36",
             input={"prompt": prompt, 'im_emb': pickle.dumps(im_emb)}
             )
+            response = requests.get(url)
+            image = Image.open(BytesIO(response.content))
+
 
+            im_emb = pickle.load(urlopen(im_emb, 'rb'))
             embs.append(im_emb)
 
             torch.save(lin_class.coef_, f'./{start_time}.pt')