AshanGimhana commited on
Commit
499cb21
·
verified ·
1 Parent(s): 9aa00c0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -5,6 +5,7 @@ os.system("pip install gradio==3.50")
5
  os.system("pip install dlib==19.24.2")
6
  os.system("pip install scikit-learn")
7
  os.system("pip install scikit-image")
 
8
 
9
 
10
  #############################################
@@ -26,7 +27,7 @@ if gpus:
26
  for gpu in gpus:
27
  tf.config.experimental.set_memory_growth(gpu, True)
28
  except RuntimeError as e:
29
- print(e) # Print error if unable to set up GPU
30
  else:
31
  print("No GPUs available.")
32
 
 
5
  os.system("pip install dlib==19.24.2")
6
  os.system("pip install scikit-learn")
7
  os.system("pip install scikit-image")
8
+ os.system("pip install tensorflow==2.11.0")
9
 
10
 
11
  #############################################
 
27
  for gpu in gpus:
28
  tf.config.experimental.set_memory_growth(gpu, True)
29
  except RuntimeError as e:
30
+ print(e)
31
  else:
32
  print("No GPUs available.")
33