Update app.py

app.py

@@ -4,8 +4,10 @@ import torch
 #from diffusers import FluxControlNetModel
 #from diffusers.pipelines import FluxControlNetPipeline
 
-from diffusers import FluxControlNetPipeline
-from diffusers import FluxControlNetModel #, FluxMultiControlNetModel
+from diffusers import DiffusionPipeline
+
+#from diffusers import FluxControlNetPipeline
+#from diffusers import FluxControlNetModel #, FluxMultiControlNetModel
 
 """
 from diffusers import DiffusionPipeline
@@ -45,7 +47,8 @@ def load_inpainting_model():
 
 # Load the model once globally to avoid repeated loading
 def load_upscaling_model():
-# Load pipeline
+    # Load pipeline
+    """
     device = "cpu"  # Explicitly use CPU
     controlnet = FluxControlNetModel.from_pretrained(
         "jasperai/Flux.1-dev-Controlnet-Upscaler",
@@ -56,6 +59,8 @@ def load_upscaling_model():
         controlnet=controlnet,
         torch_dtype=torch.float32
     ).to(device)
+    """
+    pipe = DiffusionPipeline.from_pretrained("jasperai/Flux.1-dev-Controlnet-Upscaler")    
     return pipe
 
 # Preload the model once