Upload app.py with huggingface_hub

app.py

@@ -5,6 +5,7 @@ import sys
 from pathlib import Path
 
 import gradio as gr
+import torch
 
 CURRENT_FILE = Path(__file__).resolve()
 PROJECT_ROOT = CURRENT_FILE.parents[1]
@@ -53,6 +54,45 @@ def build_gallery(result: dict) -> list[tuple[str, str]]:
     return gallery
 
 
+def get_gpu_name() -> str:
+    if not torch.cuda.is_available():
+        return "None"
+    try:
+        return torch.cuda.get_device_name(torch.cuda.current_device())
+    except Exception as exc:  # pragma: no cover - defensive runtime reporting
+        return f"Unavailable ({exc})"
+
+
+def get_runtime_status_markdown() -> str:
+    accelerator = os.environ.get("ACCELERATOR", "unknown")
+    space_id = os.environ.get("SPACE_ID", "local")
+    model_repo = os.environ.get("SYNLAYERS_MODEL_REPO", "(unset)")
+    cuda_available = torch.cuda.is_available()
+
+    lines = [
+        "## Runtime Status",
+        f"- `SPACE_ID`: `{space_id}`",
+        f"- `ACCELERATOR`: `{accelerator}`",
+        f"- `CUDA available`: `{cuda_available}`",
+        f"- `GPU device`: `{get_gpu_name()}`",
+        f"- `SYNLAYERS_MODEL_REPO`: `{model_repo}`",
+        "",
+    ]
+
+    if accelerator == "none" or not cuda_available:
+        lines.extend(
+            [
+                "This Space is not currently running with a usable CUDA GPU.",
+                "The GPU type must be chosen by the Space owner in Hugging Face `Settings -> Hardware`.",
+                "Visitors cannot switch GPUs from inside the Gradio app.",
+            ]
+        )
+    else:
+        lines.append("The CUDA runtime is available and the full SynLayers pipeline can run here.")
+
+    return "\n".join(lines)
+
+
 def run_demo(
     image_path: str,
     sample_name: str,
@@ -101,8 +141,13 @@ with gr.Blocks(title="SynLayers Real-World Demo") as demo:
 
         This Space is intended to run on GPU hardware. The first request may take time
         while model assets are loaded from Hugging Face.
+
+        GPU hardware is selected in the Hugging Face Space settings by the owner,
+        not from inside this app.
         """
     )
+    runtime_status = gr.Markdown(get_runtime_status_markdown())
+    refresh_status_button = gr.Button("Refresh Runtime Status")
 
     with gr.Row():
         with gr.Column(scale=1):
@@ -142,6 +187,11 @@ with gr.Blocks(title="SynLayers Real-World Demo") as demo:
     if examples:
         gr.Examples(examples=examples, inputs=[image_input], label="Example Images")
 
+    refresh_status_button.click(
+        fn=get_runtime_status_markdown,
+        outputs=runtime_status,
+    )
+
     run_button.click(
         fn=run_demo,
         inputs=[