Update app.py

app.py

@@ -35,17 +35,14 @@ from concurrent.futures import ProcessPoolExecutor
 
 subprocess.run('pip install flash-attn --no-build-isolation', env={'FLASH_ATTENTION_SKIP_CUDA_BUILD': "TRUE"}, shell=True)
 
-subprocess.run('snap install chromium', env={'FLASH_ATTENTION_SKIP_CUDA_BUILD': "TRUE"}, shell=True)
-
-subprocess.run('wget https://dl.google.com/linux/direct/google-chrome-stable_current_amd64.deb', env={'FLASH_ATTENTION_SKIP_CUDA_BUILD': "TRUE"}, shell=True)
-
-subprocess.run('dpkg -i google-chrome-stable_current_amd64.deb', env={'FLASH_ATTENTION_SKIP_CUDA_BUILD': "TRUE"}, shell=True)
-
-subprocess.run('apt-get update', env={'FLASH_ATTENTION_SKIP_CUDA_BUILD': "TRUE"}, shell=True)
-
-subprocess.run('apt-get install wkhtmltopdf', env={'FLASH_ATTENTION_SKIP_CUDA_BUILD': "TRUE"}, shell=True)
-
-subprocess.run('apt-get install xvfb', env={'FLASH_ATTENTION_SKIP_CUDA_BUILD': "TRUE"}, shell=True)
+env = {'FLASH_ATTENTION_SKIP_CUDA_BUILD': "TRUE"}
+
+subprocess.run('snap install chromium', env=env, shell=True)
+subprocess.run('wget https://dl.google.com/linux/direct/google-chrome-stable_current_amd64.deb', env=env, shell=True)
+subprocess.run('DEBIAN_FRONTEND=noninteractive dpkg -i google-chrome-stable_current_amd64.deb || apt-get -y -f install', env=env, shell=True)
+subprocess.run('apt-get update -y', env=env, shell=True)
+subprocess.run('apt-get install -y wkhtmltopdf', env=env, shell=True)
+subprocess.run('apt-get install -y xvfb', env=env, shell=True)