pemujo commited on
Commit
da00b58
·
1 Parent(s): 67de374

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -1,6 +1,8 @@
1
  import gradio as gr
2
  import tensorflow as tf
3
  from keras.models import load_model
 
 
4
  import segmentation_models as sm
5
  from keras.metrics import MeanIoU
6
 
 
1
  import gradio as gr
2
  import tensorflow as tf
3
  from keras.models import load_model
4
+ os.environ["TF_CPP_MIN_LOG_LEVEL"] = "2"
5
+ os.environ["SM_FRAMEWORK"] = "tf.keras"
6
  import segmentation_models as sm
7
  from keras.metrics import MeanIoU
8