update code

app.py

@@ -87,6 +87,7 @@ def install_cuda_toolkit():
         print(f"[INFO] Removed installer file: {CUDA_TOOLKIT_FILE}")
     else:
         print(f"[WARN] Installer file not found: {CUDA_TOOLKIT_FILE}")
+    print(os.listdir("/usr/local/cuda"))
     print("[INFO] CUDA 12.1 installation complete. CUDA_HOME set to /usr/local/cuda")
 
 # 🔧 Set CUDA_HOME before anything else