Update app.py

app.py

@@ -63,22 +63,25 @@ def end_session(req: gr.Request):
     shutil.rmtree(user_dir)
 
 @spaces.GPU
-def preprocess_image(image: Image.Image, 
-                    prompt: str = "",
-                    negative_prompt: str = "",
-                    style_name: str = "",
-                    num_steps: int = 25,
-                    guidance_scale: float = 5,
-                    controlnet_conditioning_scale: float = 1.0,
-                    ) -> Image.Image:
-    width, height  = image['composite'].size
-    ratio = np.sqrt(1024. * 1024. / (width * height))
+def preprocess_image(
+    image: Image.Image,
+    prompt: str = "",
+    negative_prompt: str = "",
+    style_name: str = "",
+    num_steps: int = 25,
+    guidance_scale: float = 5,
+    controlnet_conditioning_scale: float = 1.0,
+) -> Image.Image:
+    width, height = image['composite'].size
+    ratio = np.sqrt(1024.0 * 1024.0 / (width * height))
     new_width, new_height = int(width * ratio), int(height * ratio)
     image = image['composite'].resize((new_width, new_height))
     image = ImageOps.invert(image)
-    print("image:",type(image))
+    print("image:", type(image))
+    
     prompt, negative_prompt = apply_style(style_name, prompt, negative_prompt)
     print("params:", prompt, negative_prompt, style_name, num_steps, guidance_scale, controlnet_conditioning_scale)
+    
     output = pipe_control(
         prompt=prompt,
         negative_prompt=negative_prompt,
@@ -87,9 +90,11 @@ def preprocess_image(image: Image.Image,
         controlnet_conditioning_scale=controlnet_conditioning_scale,
         guidance_scale=guidance_scale,
         width=new_width,
-        height=new_height).images[0]
+        height=new_height,
+    ).images[0]
+    
     processed_image = pipeline.preprocess_image(output)
-    return (image, processed_image)
+    return processed_image  
 
 def pack_state(gs: Gaussian, mesh: MeshExtractResult) -> dict:
     return {