syntax

app.py

@@ -11,31 +11,28 @@ DEPLOYMENT_URIS = {
 
 
 def generate_image(model_selection, lora_scale, guidance_scale, prompt_strength, num_steps, prompt):
-    # deployment_uri = DEPLOYMENT_URIS[model_selection]
-    # deployment = replicate.deployments.get(deployment_uri)
-    #
-    # prediction = deployment.predictions.create(
-    #     input={
-    #         "model": "dev",
-    #         "lora_scale": lora_scale,
-    #         "num_outputs": 1,
-    #         "aspect_ratio": "1:1",
-    #         "output_format": "webp",
-    #         "guidance_scale": guidance_scale,
-    #         "output_quality": 90,
-    #         "prompt_strength": prompt_strength,
-    #         "extra_lora_scale": 1,
-    #         "num_inference_steps": num_steps,
-    #         "prompt": prompt
-    #     }
-    # )
-    
-    # prediction.wait()
-    # output = prediction.output
-    # image_url = output[0] if output else None
+    deployment_uri = DEPLOYMENT_URIS[model_selection]
+    deployment = replicate.deployments.get(deployment_uri)
 
-    time.sleep(20)
-    image_url = "https://external-content.duckduckgo.com/iu/?u=https%3A%2F%2Fi1.sndcdn.com%2Fartworks-000207677710-z7mg8u-t500x500.jpg&f=1&nofb=1&ipt=54d62f19e1816c166bec632275277c4118a940853b2cd93d1c673ea487d9dfee&ipo=images"
+    prediction = deployment.predictions.create(
+        input={
+            "model": "dev",
+            "lora_scale": lora_scale,
+            "num_outputs": 1,
+            "aspect_ratio": "1:1",
+            "output_format": "webp",
+            "guidance_scale": guidance_scale,
+            "output_quality": 90,
+            "prompt_strength": prompt_strength,
+            "extra_lora_scale": 1,
+            "num_inference_steps": num_steps,
+            "prompt": prompt
+        }
+    )
+    
+    prediction.wait()
+    output = prediction.output
+    image_url = output[0] if output else None
 
     return image_url