Update app.py

app.py

@@ -5,21 +5,10 @@ from loguru import logger
 import random
 from transformers import pipeline
 import torch
-import apt
-import sys
-pkg_name = "xvfb"
-cache = apt.cache.Cache()
-cache.update()
-cache.open()
-pkg = cache[pkg_name]
-if pkg.is_installed:
-    print ("{pkg_name} already installed".format(pkg_name=pkg_name))
-else:
-    pkg.mark_install()
-    try:
-        cache.commit()
-    except Exception as arg:
-        print (sys.stderr, "Sorry, package installation failed [{err}]".format(err=str(arg)))
+import subprocess
+subprocess.Popen( 'sudo apt-get -y install xvfb',
+                   shell=True,
+                   stdin=subprocess.PIPE ).communicate()
 # from xvfbwrapper import Xvfb
 # vdisplay = Xvfb()
 # vdisplay.start()