Update app.py

app.py

@@ -52,18 +52,9 @@ def next_image():
             prompt = calibrate_prompts.pop(0)
             print(prompt)
 
-            image = pipe(
-            prompt=prompt,
-            height=1024,
-            width=1024,
-            num_inference_steps=8,
-            guidance_scale=0,
-            ip_adapter_emb=torch.zeros(1, 1, 1280, device=DEVICE, dtype=torch.float16),
-            ).images
-
-
-            pooled_embeds, _ = pipe.encode_image(
-                image[0], DEVICE, 1, output_hidden_state
+            image, pooled_embeds = replicate.run(
+            "rynmurdock/zahir:43177e0594f3bc2e3560170ff0ffb6d1cacdddda1be25fbcd4348ef02b0b7d0f",
+            input={"prompt": prompt, 'im_emg': pickle.dumps(im_emb)}
             )
 
             embs.append(pooled_embeds)