Update app.py

app.py

@@ -13,6 +13,22 @@ def run_command(command):
         exit(1)
     return output.decode('utf-8')
 
+# Download CUDA installer
+download_command = "wget https://developer.download.nvidia.com/compute/cuda/12.2.0/local_installers/cuda_12.2.0_535.54.03_linux.run"
+result = run_command(download_command)
+if result is None:
+    print("Failed to download CUDA installer.")
+    exit(1)
+
+# Run CUDA installer
+install_command = "sudo sh cuda_12.2.0_535.54.03_linux.run"
+result = run_command(install_command)
+if result is None:
+    print("Failed to run CUDA installer.")
+    exit(1)
+
+print("CUDA installation process completed.")
+
 def install_packages():
     
     # Clone the repository with submodules