Update app.py

app.py

@@ -11,6 +11,20 @@ from huggingface_hub import hf_hub_download
 from aura_sr import AuraSR
 from gradio_imageslider import ImageSlider
 
+device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
+
+# --- # Device and CUDA Setup Check ---
+print("CUDA_VISIBLE_DEVICES=", os.environ.get("CUDA_VISIBLE_DEVICES"))
+print("torch.__version__ =", torch.__version__)
+print("torch.version.cuda =", torch.version.cuda)
+print("cuda available:", torch.cuda.is_available())
+print("cuda device count:", torch.cuda.device_count())
+if torch.cuda.is_available():
+    print("current device:", torch.cuda.current_device())
+    print("device name:", torch.cuda.get_device_name(torch.cuda.current_device()))
+
+print("Using device:", device)
+
 # --- Main Model Initialization ---
 MAX_SEED = np.iinfo(np.int32).max
 pipe = FluxKontextPipeline.from_pretrained("black-forest-labs/FLUX.1-Kontext-dev", torch_dtype=torch.bfloat16).to("cuda")
@@ -182,4 +196,4 @@ with gr.Blocks(css=css, theme="bethecloud/storj_theme") as demo:
         outputs=[input_image]
     )
 
-demo.launch(mcp_server=True)
+demo.launch(share=True, mcp_server=True, ssr_mode=False, show_error=True)