Upload folder using huggingface_hub

app.py

@@ -7,15 +7,18 @@ import torch
 import gradio as gr
 
 INFO = (
-    f"python={sys.version.split()[0]} ({sys.version.split('(')[0].strip()})\n"
-    f"torch={torch.__version__}\n"
-    f"torch.version.cuda={torch.version.cuda}\n"
-    f"cxx11_abi={torch._C._GLIBCXX_USE_CXX11_ABI}\n"
-    f"cuda_available={torch.cuda.is_available()}\n"
-    f"TORCH_CUDA_ARCH_LIST={os.environ.get('TORCH_CUDA_ARCH_LIST')!r}\n"
-    f"machine={os.uname().machine}\n"
+    f"PROBE_PYTHON={sys.version.split()[0]}\n"
+    f"PROBE_TORCH={torch.__version__}\n"
+    f"PROBE_CUDA={torch.version.cuda}\n"
+    f"PROBE_CXX11ABI={torch._C._GLIBCXX_USE_CXX11_ABI}\n"
+    f"PROBE_TORCH_CUDA_ARCH_LIST={os.environ.get('TORCH_CUDA_ARCH_LIST')!r}\n"
+    f"PROBE_MACHINE={os.uname().machine}\n"
 )
 
+print("===PROBE-START===", flush=True)
+print(INFO, flush=True)
+print("===PROBE-END===", flush=True)
+
 
 @spaces.GPU(duration=60)
 def probe():