salmanmapkar commited on
Commit
75497db
·
1 Parent(s): a01009d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -20,9 +20,9 @@ else:
20
  cache.commit()
21
  except Exception, arg:
22
  print >> sys.stderr, "Sorry, package installation failed [{err}]".format(err=str(arg))
23
- from xvfbwrapper import Xvfb
24
- vdisplay = Xvfb()
25
- vdisplay.start()
26
  session_token = os.environ.get('SessionToken')
27
 
28
  device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
 
20
  cache.commit()
21
  except Exception, arg:
22
  print >> sys.stderr, "Sorry, package installation failed [{err}]".format(err=str(arg))
23
+ # from xvfbwrapper import Xvfb
24
+ # vdisplay = Xvfb()
25
+ # vdisplay.start()
26
  session_token = os.environ.get('SessionToken')
27
 
28
  device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")