Update app.py

app.py

@@ -66,20 +66,38 @@ async def generate_image(
     
     return image, seed
 
+def get_upscale_finegrain(prompt, img_path, upscale_factor):
+    client = Client("finegrain/finegrain-image-enhancer")
+    result = client.predict(
+        input_image=handle_file(img_path),
+        prompt=prompt,
+        negative_prompt="",
+        seed=42,
+        upscale_factor=upscale_factor,
+        controlnet_scale=0.6,
+        controlnet_decay=1,
+        condition_scale=6,
+        tile_width=112,
+        tile_height=144,
+        denoise_strength=0.35,
+        num_inference_steps=18,
+        solver="DDIM",
+        api_name="/process"
+    )
+    print(result)
+    return result[1]
+
 async def upscale_image(image, upscale_factor):
     try:
-        result = await client.text_to_image(
+        result = get_upscale_finegrain(
             prompt="",
-            height=image.height * upscale_factor,
-            width=image.width * upscale_factor,
-            guidance_scale=3.5,
-            num_inference_steps=18,
-            model="finegrain/finegrain-image-enhancer"
+            img_path=image,
+            upscale_factor=upscale_factor
         )
     except Exception as e:
         raise gr.Error(f"Error in {e}")
     
-    return result[1]
+    return result
 
 async def gen(
     prompt:str,