Upload app.py with huggingface_hub

app.py

@@ -1,7 +1,7 @@
 import os, sys, subprocess, platform
-import gradio as gr
 
-def probe():
+# Probe at import so it lands in run logs regardless of UI/SSR state
+def _probe():
     lines = []
     lines.append(f"python: {sys.version}")
     lines.append(f"platform: {platform.platform()}")
@@ -9,31 +9,34 @@ def probe():
         import torch
         lines.append(f"torch: {torch.__version__}")
         lines.append(f"torch.version.cuda: {torch.version.cuda}")
-        lines.append(f"cuda available: {torch.cuda.is_available()}")
     except Exception as e:
         lines.append(f"torch import error: {e!r}")
-    # nvcc
-    for cmd in (["nvcc", "--version"], ["which", "nvcc"]):
+    for cmd in (["nvcc", "--version"], ["which", "nvcc"], ["ls", "/usr/local/cuda/bin"]):
         try:
             out = subprocess.run(cmd, capture_output=True, text=True, timeout=20)
-            lines.append(f"$ {' '.join(cmd)}\n{out.stdout}{out.stderr}")
+            lines.append(f"$ {' '.join(cmd)} (rc={out.returncode})\n{out.stdout}{out.stderr}")
         except Exception as e:
             lines.append(f"{' '.join(cmd)} error: {e!r}")
-    lines.append("CUDA_HOME=" + str(os.environ.get("CUDA_HOME")))
-    lines.append("PATH=" + str(os.environ.get("PATH")))
-    # transformers version
+    # look for nvcc from pip nvidia-cuda-nvcc
     try:
-        import transformers
-        lines.append(f"transformers: {transformers.__version__}")
+        import glob
+        cands = glob.glob("/usr/local/lib/python3.10/site-packages/nvidia/**/nvcc", recursive=True)
+        cands += glob.glob("/usr/local/lib/python3.10/site-packages/nvidia/cuda_nvcc/bin/*", recursive=True)
+        lines.append("pip nvcc candidates: " + str(cands))
     except Exception as e:
-        lines.append(f"transformers: {e!r}")
+        lines.append(f"glob err {e!r}")
+    lines.append("CUDA_HOME=" + str(os.environ.get("CUDA_HOME")))
     return "\n".join(lines)
 
+PROBE = _probe()
+print("=========== PROBE START ===========")
+print(PROBE)
+print("=========== PROBE END ===========", flush=True)
+
+import gradio as gr
+
 with gr.Blocks() as demo:
     gr.Markdown("# Probe")
-    out = gr.Textbox(label="env", lines=30)
-    btn = gr.Button("probe")
-    btn.click(probe, None, out)
-    demo.load(probe, None, out)
+    gr.Textbox(value=PROBE, label="env", lines=30)
 
-demo.launch()
+demo.launch(ssr_mode=False)