Upload app.py with huggingface_hub

app.py

@@ -1,21 +1,14 @@
 import spaces
 import torch
 import gradio as gr
-
-# Import the heavy deps at module scope to see which (if any) poisons the fork.
-import transformers  # noqa
-import torch_geometric  # noqa
-print("transformers", transformers.__version__, "pyg", torch_geometric.__version__)
-print("cuda initialized in parent?", torch.cuda.is_initialized())
+import transformers  # noqa  -- test transformers alone
+print("PROBE_MARK transformers", transformers.__version__, "cuda_init_parent=", torch.cuda.is_initialized(), flush=True)
 
 @spaces.GPU(duration=30)
 def probe():
     x = torch.randn(4, 4, device="cuda")
-    return f"OK: {(x @ x).sum().item():.3f} on {torch.cuda.get_device_name(0)}"
+    return f"OK: {(x @ x).sum().item():.3f}"
 
 with gr.Blocks() as demo:
-    out = gr.Textbox(label="probe")
-    btn = gr.Button("probe")
-    btn.click(probe, outputs=out, api_name="probe")
-
+    gr.Button("probe").click(probe, outputs=gr.Textbox(), api_name="probe")
 demo.launch()