fixed install bug adding nvidia repo

app.py

@@ -34,18 +34,21 @@ def install_cuda_if_gpu_detected():
     if torch.cuda.is_available():
         print("GPU detected! Proceeding with installation...")
         try:
-            # Update package list
+            # Add NVIDIA's package repositories
+            subprocess.run(['apt-key', 'adv', '--fetch-keys', 'https://developer.download.nvidia.com/compute/cuda/repos/debian11/x86_64/3bf863cc.pub'], check=True)
+            subprocess.run(['add-apt-repository', 'deb https://developer.download.nvidia.com/compute/cuda/repos/debian11/x86_64/ /'], check=True)
             subprocess.run(['apt-get', 'update'], check=True)
-            
+
             # Install CUDA dependencies
             subprocess.run(['apt-get', 'install', '-y', 'libcudnn8', 'libcudnn8-dev'], check=True)
-            
+
             print("Installation complete.")
         except subprocess.CalledProcessError as e:
             print(f"An error occurred during installation: {e}")
     else:
         print("No GPU detected. Skipping installation.")
 
+
 # Run the function
 install_cuda_if_gpu_detected()