Update src/streamlit_app.py

src/streamlit_app.py

@@ -2,6 +2,15 @@ import os
 import shutil
 import streamlit as st
 
+import torch
+print("CUDA available:", torch.cuda.is_available())
+print("Device count:", torch.cuda.device_count())
+if torch.cuda.is_available():
+    print("GPU name:", torch.cuda.get_device_name(0))
+else:
+    print("Running on CPU")
+
+
 # ==========================================================
 # ✅ Page Configuration (must be first Streamlit command)
 # ==========================================================