Jayem-11 commited on
Commit
09767df
·
1 Parent(s): 45b9135

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -1,6 +1,8 @@
1
- import os
2
  import cv2
3
  from PIL import Image
 
 
 
4
  import segmentation_models as sm
5
  import numpy as np
6
  from matplotlib import pyplot as plt
 
 
1
  import cv2
2
  from PIL import Image
3
+
4
+ import os
5
+ os.environ["SM_FRAMEWORK"] = "tf.keras"
6
  import segmentation_models as sm
7
  import numpy as np
8
  from matplotlib import pyplot as plt