Create app.py

app.py

@@ -0,0 +1,18 @@
+import streamlit as st
+import torch
+
+def detect_gpu():
+    if torch.cuda.is_available():
+        device = torch.cuda.get_device_name(0)
+        device_memory = torch.cuda.get_device_properties(0).total_memory
+        return f"GPU detected: {device}\nMemory: {device_memory} bytes"
+    else:
+        return "No GPU detected."
+
+def main():
+    st.title("GPU Detection Demo")
+    result = detect_gpu()
+    st.write(result)
+
+if __name__ == "__main__":
+    main()