Update app.py

app.py

@@ -297,13 +297,17 @@ model_path = snapshot_download(
 )
 
 # Load pipeline
+print('controlnet enters')
 controlnet = FluxControlNetModel.from_pretrained(
     "jasperai/Flux.1-dev-Controlnet-Upscaler", torch_dtype=torch.bfloat16
 ).to(device)
+print('controlnet exits')
+print('pipe enters')
 pipe = FluxControlNetPipeline.from_pretrained(
     model_path, controlnet=controlnet, torch_dtype=torch.bfloat16
 )
 pipe.to(device)
+print('pipe exits')
 
 MAX_SEED = 1000000
 MAX_PIXEL_BUDGET = 1024 * 1024