Update app.py

app.py

@@ -27,6 +27,9 @@ start_time = time.time()
 # TODO add to state instead of shared across all
 glob_idx = 0
 
+
+deployment = replicate.deployments.get("rynmurdock/zahir-deployment")
+
 def next_image(embs, ys, calibrate_prompts):
     global glob_idx
     glob_idx = glob_idx + 1
@@ -44,10 +47,17 @@ def next_image(embs, ys, calibrate_prompts):
             prompt = calibrate_prompts.pop(0)
             print(prompt)
 
-            output = replicate.run(
-            "rynmurdock/zahir:42c58addd49ab57f1e309f0b9a0f271f483bbef0470758757c623648fe989e42",
-            input={"prompt": prompt,}
+            prediction = deployment.predictions.create(
+              input={"prompt": prompt,}
             )
+            prediction.wait()
+            output = prediction.output
+
+            # output = replicate.run(
+            # "rynmurdock/zahir:42c58addd49ab57f1e309f0b9a0f271f483bbef0470758757c623648fe989e42",
+            # input={"prompt": prompt,}
+            # )
+            
             response = requests.get(output['file1'])
             image = Image.open(BytesIO(response.content))
 
@@ -86,11 +96,17 @@ def next_image(embs, ys, calibrate_prompts):
 
             im_emb_st = str(im_emb[0].cpu().detach().tolist())[1:-1]
             
-
-            output = replicate.run(
-            "rynmurdock/zahir:42c58addd49ab57f1e309f0b9a0f271f483bbef0470758757c623648fe989e42",
-            input={"prompt": prompt, 'im_emb': im_emb_st}
+            prediction = deployment.predictions.create(
+              input={"prompt": prompt, 'im_emb': im_emb_st}
             )
+            prediction.wait()
+            output = prediction.output
+            
+            # output = replicate.run(
+            # "rynmurdock/zahir:42c58addd49ab57f1e309f0b9a0f271f483bbef0470758757c623648fe989e42",
+            # input={"prompt": prompt, 'im_emb': im_emb_st}
+            # )
+            
             response = requests.get(output['file1'])
             image = Image.open(BytesIO(response.content))