Update app.py

app.py

@@ -35,12 +35,17 @@ def style_transfer(input_image_path, style_image_path, prompt_det):
         "seed": 42,
         "guidance_scale": 5
     }
-    output = replicate.run(
-        # "prakharsaxena24/2d-to-real-style:c0e1e612a11a13d1d57a6d647e7665ad850bc73715337c1f499bb7b52404c35a",
-        "prakharsaxena24/2d-to-real-style:fef4d74fb7d11df35aa4bbdf3d8671b4d0352464dc67b169968393c657ab6038",
-        input=input
+    # output = replicate.run(
+    #     # "prakharsaxena24/2d-to-real-style:fef4d74fb7d11df35aa4bbdf3d8671b4d0352464dc67b169968393c657ab6038",
+    #     input=input
+    # )
+    # return output[0]
+    deployment = replicate.deployments.get("2clicksmedia/my-app-photorealism")
+    prediction = deployment.predictions.create(
+      input=input
     )
-    return output[0]
+    prediction.wait()
+    return prediction.output[0]
 
 def upscale_image(image_path, prompt_det):
     input = {
@@ -50,12 +55,18 @@ def upscale_image(image_path, prompt_det):
         "negative_prompt": "hands, fingers, feet, legs, shoes",
 
     }
-    output = replicate.run(
-        "philz1337x/clarity-upscaler:eba39f520856d5e61a8ad56fd57f97be2fa30de65e29d8e94db5209a1827cd59",
-        # "prakharsaxena24/calrity-upscaler-private",
-        input=input)
+    # output = replicate.run(
+    #     "philz1337x/clarity-upscaler:eba39f520856d5e61a8ad56fd57f97be2fa30de65e29d8e94db5209a1827cd59",
+    #     # "prakharsaxena24/calrity-upscaler-private",
+    #     input=input)
     
-    return output[0]
+    # return output[0]
+    deployment = replicate.deployments.get("2clicksmedia/upscaler")
+    prediction = deployment.predictions.create(
+      input={"image": "..."}
+    )
+    prediction.wait()
+    return prediction.output[0]
 
 def get_keyword_prompt(image_url):
     client = OpenAI()