salmanmapkar commited on
Commit
c9d1588
·
1 Parent(s): 4830db0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -5,8 +5,8 @@ from loguru import logger
5
  import random
6
  from transformers import pipeline
7
  import torch
8
- from selenium_profiles.utils.installer import install_chromedriver
9
- install_chromedriver()
10
  session_token = os.environ.get('SessionToken')
11
 
12
  device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
 
5
  import random
6
  from transformers import pipeline
7
  import torch
8
+ import chromedriver_autoinstaller
9
+ chromedriver_autoinstaller.install()
10
  session_token = os.environ.get('SessionToken')
11
 
12
  device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")