aixen commited on
Commit
941731c
1 Parent(s): 1feca79

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +10 -0
app.py CHANGED
@@ -1,6 +1,16 @@
1
  import importlib
2
 
3
  # Check if transformers is already installed
 
 
 
 
 
 
 
 
 
 
4
  try:
5
  importlib.import_module('transformers')
6
  print("transformers library is already installed")
 
1
  import importlib
2
 
3
  # Check if transformers is already installed
4
+
5
+
6
+ try:
7
+ importlib.import_module('tensorflow')
8
+ print("TensorFlow library is already installed")
9
+ except ImportError:
10
+ # TensorFlow is not installed, so install it
11
+ import subprocess
12
+ subprocess.check_call(["pip", "install", "tensorflow==2.0"])
13
+ print("TensorFlow library has been installed")
14
  try:
15
  importlib.import_module('transformers')
16
  print("transformers library is already installed")