Upload app.py with huggingface_hub

app.py

@@ -1,8 +1,15 @@
+import os
+# torch_geometric / pyg-lib can call cuInit at import; try to suppress.
+os.environ.setdefault("TORCH_GEOMETRIC_DISABLE_EXTENSIONS", "1")
 import spaces
 import torch
 import gradio as gr
-import transformers  # noqa  -- test transformers alone
-print("PROBE_MARK transformers", transformers.__version__, "cuda_init_parent=", torch.cuda.is_initialized(), flush=True)
+try:
+    import torch_geometric  # noqa  -- test pyg alone
+    _pyg = torch_geometric.__version__
+except Exception as e:
+    _pyg = f"import-failed: {e!r}"
+print("PROBE_MARK pyg", _pyg, "cuda_init_parent=", torch.cuda.is_initialized(), flush=True)
 
 @spaces.GPU(duration=30)
 def probe():