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

Update app.py

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