sanjay-subramanya commited on
Commit
d38fb83
1 Parent(s): 35d0da4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -7,6 +7,8 @@ import segmentation_models as sm
7
  from matplotlib import pyplot as plt
8
 
9
  import tensorflow as tf
 
 
10
  from tensorflow.keras import backend as K
11
  from tensorflow.keras.models import load_model
12
 
 
7
  from matplotlib import pyplot as plt
8
 
9
  import tensorflow as tf
10
+ %env SM_FRAMEWORK=tf.keras
11
+
12
  from tensorflow.keras import backend as K
13
  from tensorflow.keras.models import load_model
14