Update app.py

app.py

@@ -58,13 +58,13 @@ def next_image():
             print(prompt)
 
             image, pooled_embeds = replicate.run(
-            "rynmurdock/zahir:e64023db020f28b77d407c8954c82530df96679ae7d97e302c7ed9d9e68b851a",
+            "rynmurdock/zahir:49ebb1916c4baae35884ebfa16b092cf45d086c1913b53f62bb07d575cdbe683",
             input={"prompt": prompt,}
             )
             response = requests.get(url)
             image = Image.open(BytesIO(response.content))
 
-            embs.append(pickle.load(urlopen(pooled_embeim_embds, 'rb')))
+            embs.append(torch.tensor(urlopen(pooled_embeds).read().decode('utf-8').split(', ')))
             return image
         else:
             print('######### Roaming #########')
@@ -98,14 +98,14 @@ def next_image():
             print(prompt)
 
             image, im_emb = replicate.run(
-            "rynmurdock/zahir:e64023db020f28b77d407c8954c82530df96679ae7d97e302c7ed9d9e68b851a",
+            "rynmurdock/zahir:49ebb1916c4baae35884ebfa16b092cf45d086c1913b53f62bb07d575cdbe683",
             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'))
+            im_emb = torch.tensor(urlopen(im_emb).read().decode('utf-8').split(', '))
             embs.append(im_emb)
 
             torch.save(lin_class.coef_, f'./{start_time}.pt')
@@ -166,7 +166,7 @@ with gr.Blocks(css=css) as demo:
         [img]
         )
     with gr.Row():
-        b4 = gr.Button(value='Start')
+        b4 = gr.Button(value='Start'9bc92ec26fa95fe2ee8e15501ad9085bbb4013b42be9cd2b48c8e5086bc4ae83)
         b4.click(start,
                  [b4],
                  [b1, b2, b3, b4, img,])