jeysshon commited on
Commit
953654d
1 Parent(s): b4d408c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -0
app.py CHANGED
@@ -2,6 +2,9 @@
2
  import gradio as gr
3
  import tensorflow as tf
4
 
 
 
 
5
  path_to_model = "./modelo_jeysshon_iaderm.h5"
6
 
7
  model = tf.keras.models.load_model(path_to_model)
 
2
  import gradio as gr
3
  import tensorflow as tf
4
 
5
+ # TensorFlow Setup and Warnings
6
+ tf.config.set_visible_devices([], 'GPU') # Disable GPU for now
7
+
8
  path_to_model = "./modelo_jeysshon_iaderm.h5"
9
 
10
  model = tf.keras.models.load_model(path_to_model)