Update app.py

app.py

@@ -38,6 +38,20 @@ def style_transfer(input_image_path, style_image_path, revised_prompt):
     )
     return output[0]
 
+def upscale_image(image_path):
+    input = {
+        "image": image_path,
+        "prompt": "candid photo, high iso, phone camera, grainy <lora:more_details:0.5> , symmetric hands" ,
+        "scale_factor": 2,
+
+    }
+    output = replicate.run(
+        "philz1337x/clarity-upscaler:eba39f520856d5e61a8ad56fd57f97be2fa30de65e29d8e94db5209a1827cd59",
+        input=input)
+    
+    return output[0]
+
+
 
 
 def infer(text,title):
@@ -46,10 +60,12 @@ def infer(text,title):
      Please make sure not to include text in the image."""
     image_url_openai, revised_prompt = generate_image_openai(prompt)
     style_image_url = style_transfer(image_url_openai, f'./Style.png',revised_prompt)
+    upscaled_image_url = upscale_image(style_image_url)
 
     response_dalle = requests.get(image_url_openai)
     dalle_img = Image.open(BytesIO(response_dalle.content))
-    response = requests.get(style_image_url)
+    
+    response = requests.get(upscaled_image_url)
     img = Image.open(BytesIO(response.content))
 
     return dalle_img, img