SineTheta commited on
Commit
7aa55e8
·
1 Parent(s): e04912d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -2
app.py CHANGED
@@ -4,13 +4,16 @@ import re
4
  import subprocess
5
  import sys
6
 
7
- subprocess.check_call([sys.executable, "-m", "pip", "uninstall", 'tensorflow'])
8
  subprocess.check_call([sys.executable, "-m", "pip", "install", 'tensorflow==2.10.0'])
9
  subprocess.check_call([sys.executable, "-m", "pip", "install", 'tensorflow-gpu'])
10
- subprocess.check_call([sys.executable, "-m", "pip", "uninstall", 'transformers'])
11
  subprocess.check_call([sys.executable, "-m", "pip", "install", 'transformers'])
12
  subprocess.check_call([sys.executable, "-m", "pip", "install", 'keras==2.10.0'])
 
13
 
 
 
14
 
15
  from transformers import pipeline
16
 
 
4
  import subprocess
5
  import sys
6
 
7
+ # subprocess.check_call([sys.executable, "-m", "pip", "uninstall", 'tensorflow'])
8
  subprocess.check_call([sys.executable, "-m", "pip", "install", 'tensorflow==2.10.0'])
9
  subprocess.check_call([sys.executable, "-m", "pip", "install", 'tensorflow-gpu'])
10
+ # subprocess.check_call([sys.executable, "-m", "pip", "uninstall", 'transformers'])
11
  subprocess.check_call([sys.executable, "-m", "pip", "install", 'transformers'])
12
  subprocess.check_call([sys.executable, "-m", "pip", "install", 'keras==2.10.0'])
13
+ subprocess.check_call([sys.executable, "-m", "pip", "install", 'pytorch-pretrained-bert'])
14
 
15
+ import pytorch_pretrained_bert as ppb
16
+ assert 'bert-large-cased' in ppb.modeling.PRETRAINED_MODEL_ARCHIVE_MAP
17
 
18
  from transformers import pipeline
19