salmanmapkar commited on
Commit
ea40b87
1 Parent(s): f6a91fa

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -5,8 +5,9 @@ from loguru import logger
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")
 
5
  import random
6
  from transformers import pipeline
7
  import torch
8
+ from webdriver_manager.chrome import ChromeDriverManager
9
+ # import chromedriver_autoinstaller
10
+ # chromedriver_autoinstaller.install()
11
  session_token = os.environ.get('SessionToken')
12
 
13
  device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")