SineTheta commited on
Commit
4e27c2b
·
1 Parent(s): 3bd400c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -6,6 +6,7 @@ import sys
6
 
7
  subprocess.check_call([sys.executable, "-m", "pip", "uninstall", 'tensorflow'])
8
  subprocess.check_call([sys.executable, "-m", "pip", "install", 'tensorflow==2.4.1'])
 
9
  subprocess.check_call([sys.executable, "-m", "pip", "uninstall", 'transformers'])
10
  subprocess.check_call([sys.executable, "-m", "pip", "install", 'transformers'])
11
  subprocess.check_call([sys.executable, "-m", "pip", "install", 'keras==2.4.3'])
 
6
 
7
  subprocess.check_call([sys.executable, "-m", "pip", "uninstall", 'tensorflow'])
8
  subprocess.check_call([sys.executable, "-m", "pip", "install", 'tensorflow==2.4.1'])
9
+ subprocess.check_call([sys.executable, "-m", "pip", "install", 'tensorflow-gpu= 2.4.1'])
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.4.3'])