fradinho commited on
Commit
6c68373
·
1 Parent(s): 28a496b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -7
app.py CHANGED
@@ -7,15 +7,12 @@ import tensorflow
7
  import tensorflow as tf
8
  from tensorflow.keras import backend as K
9
  from tensorflow.keras.models import Model
10
- import segmentation_models as sm
11
- from tensorflow import keras
12
-
13
 
14
- sm.set_framework('tf.keras')
15
-
16
- sm.framework()
17
 
18
- from tensorflow.keras.applications import EfficientNetB7
19
 
20
  def jacard(y_true, y_pred):
21
  y_true_c = K.flatten(y_true)
 
7
  import tensorflow as tf
8
  from tensorflow.keras import backend as K
9
  from tensorflow.keras.models import Model
10
+ import os
11
+ os.environ["SM_FRAMEWORK"] = "tf.keras"
 
12
 
13
+ from tensorflow import keras
14
+ import segmentation_models as sm
 
15
 
 
16
 
17
  def jacard(y_true, y_pred):
18
  y_true_c = K.flatten(y_true)