Update app.py

app.py

@@ -573,7 +573,6 @@ def infer_upscale(
         
         input_image, w_original, h_original, was_resized = process_input(input_image, upscale_factor)
 
-
         # rescale with upscale factor
         w, h = input_image.size
         control_image = input_image.resize((w * upscale_factor, h * upscale_factor))
@@ -583,7 +582,12 @@ def infer_upscale(
         gr.Info("Upscaling image...")
         # 모든 텐서를 동일한 디바이스로 이동
         pipe_upscale.to(device)
-        control_image = torch.from_numpy(np.array(control_image)).permute(2, 0, 1).float().to(device)
+        
+        # Ensure the image is in RGB format
+        if control_image.mode != 'RGB':
+            control_image = control_image.convert('RGB')
+        
+        control_image = torch.from_numpy(np.array(control_image)).permute(2, 0, 1).float().to(device) / 255.0
         
         image = pipe_upscale(
             prompt="",
@@ -602,10 +606,12 @@ def infer_upscale(
         # resize to target desired size
         image = image.resize((w_original * upscale_factor, h_original * upscale_factor))
         
-
+        return image, seed
 
     except Exception as e:
         print(f"Error in infer_upscale: {str(e)}")
+        import traceback
+        traceback.print_exc()
         return None, seed