Update app.py

app.py

@@ -17,6 +17,7 @@ from pipeline_flux_control_removal import FluxControlRemovalPipeline
 torch.set_grad_enabled(False)
 os.environ['GRADIO_TEMP_DIR'] = './tmp'
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
+print(device)
 image_path = mask_path = None
 image_examples = [...] 
 image_path = mask_path =None