Create test.py

Testing if the environ is on and check GPU

test.py

@@ -0,0 +1,19 @@
+import streamlit as st
+import subprocess
+import torch
+
+st.title("Hugging Face Space GPU Info")
+
+# Check if GPU is available
+gpu_available = torch.cuda.is_available()
+
+if gpu_available:
+    gpu_name = torch.cuda.get_device_name(0)
+    st.write(f"GPU available: {gpu_name}")
+else:
+    st.write("No GPU available")
+
+# Ping the server
+st.title("Server Ping")
+ping_output = subprocess.check_output(["ping", "-c", "4", "google.com"])
+st.write("Ping output:\n", ping_output.decode("utf-8"))