update code

app.py

@@ -82,7 +82,11 @@ def install_cuda_toolkit():
 
     # Optional: set architecture list for compilation (Ampere and Ada)
     os.environ["TORCH_CUDA_ARCH_LIST"] = "8.0;8.6;8.9"
-
+    if os.path.exists(CUDA_TOOLKIT_FILE):
+        os.remove(CUDA_TOOLKIT_FILE)
+        print(f"[INFO] Removed installer file: {CUDA_TOOLKIT_FILE}")
+    else:
+        print(f"[WARN] Installer file not found: {CUDA_TOOLKIT_FILE}")
     print("[INFO] CUDA 12.1 installation complete. CUDA_HOME set to /usr/local/cuda")
 
 # 🔧 Set CUDA_HOME before anything else