Update app.py

app.py

@@ -16,6 +16,23 @@ import gradio as gr
 import torch
 from transformers import AutoTokenizer, AutoModelForCausalLM
 
+def _print_runtime_env():
+    import platform, torch, os
+    print("======== RUNTIME ========")
+    print("Python:", platform.python_version())
+    print("PyTorch:", torch.__version__)
+    print("CUDA available:", torch.cuda.is_available())
+    if torch.cuda.is_available():
+        print("CUDA device count:", torch.cuda.device_count())
+        print("CUDA device name:", torch.cuda.get_device_name(0))
+        print("torch.version.cuda:", getattr(torch.version, "cuda", None))
+    print("MPS available (Apple):", torch.backends.mps.is_available() if hasattr(torch.backends, "mps") else False)
+    print("HF_HOME:", os.getenv("HF_HOME"))
+    print("=========================")
+
+_print_runtime_env()
+
+
 # ---------------------------
 # Config & Secrets
 # ---------------------------