Update app.py

app.py

@@ -5,13 +5,8 @@ from loguru import logger
 import random
 from transformers import pipeline
 import torch
-import subprocess
-subprocess.Popen( 'apt -y install xvfb',
-                   shell=True,
-                   stdin=subprocess.PIPE ).communicate()
-# from xvfbwrapper import Xvfb
-# vdisplay = Xvfb()
-# vdisplay.start()
+from selenium_profiles.utils.installer import install_chromedriver
+install_chromedriver()
 session_token = os.environ.get('SessionToken')      
 
 device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")