Update app.py

app.py

@@ -57,14 +57,14 @@ def next_image():
             prompt = calibrate_prompts.pop(0)
             print(prompt)
 
-            image, pooled_embeds = replicate.run(
+            image = replicate.run(
             "rynmurdock/zahir:49ebb1916c4baae35884ebfa16b092cf45d086c1913b53f62bb07d575cdbe683",
             input={"prompt": prompt,}
             )
-            response = requests.get(image)
+            response = requests.get(image['file1'])
             image = Image.open(BytesIO(response.content))
 
-            embs.append(torch.tensor(urlopen(pooled_embeds).read().decode('utf-8').split(', ')))
+            embs.append(torch.tensor(urlopen(image['file2']).read().decode('utf-8').split(', ')))
             return image
         else:
             print('######### Roaming #########')
@@ -97,15 +97,15 @@ def next_image():
             prompt= 'an image' if glob_idx % 2 == 0 else rng_prompt
             print(prompt)
 
-            image, im_emb = replicate.run(
+            image = replicate.run(
             "rynmurdock/zahir:49ebb1916c4baae35884ebfa16b092cf45d086c1913b53f62bb07d575cdbe683",
             input={"prompt": prompt, 'im_emb': pickle.dumps(im_emb)}
             )
-            response = requests.get(image)
+            response = requests.get(image['file1'])
             image = Image.open(BytesIO(response.content))
 
 
-            im_emb = torch.tensor(urlopen(im_emb).read().decode('utf-8').split(', '))
+            im_emb = torch.tensor(urlopen(image['file2']).read().decode('utf-8').split(', '))
             embs.append(im_emb)
 
             torch.save(lin_class.coef_, f'./{start_time}.pt')