Create app.py

app.py

@@ -0,0 +1,13 @@
+import streamlit as st
+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")