Update backend/main.py
Browse files- backend/main.py +3 -0
backend/main.py
CHANGED
|
@@ -14,6 +14,9 @@ import cv2 # type: ignore
|
|
| 14 |
import joblib # type: ignore
|
| 15 |
import tensorflow as tf # type: ignore
|
| 16 |
import segmentation_models as sm # type: ignore
|
|
|
|
|
|
|
|
|
|
| 17 |
|
| 18 |
|
| 19 |
# -----------------------------
|
|
|
|
| 14 |
import joblib # type: ignore
|
| 15 |
import tensorflow as tf # type: ignore
|
| 16 |
import segmentation_models as sm # type: ignore
|
| 17 |
+
import os
|
| 18 |
+
os.environ["SM_FRAMEWORK"] = "tf.keras"
|
| 19 |
+
|
| 20 |
|
| 21 |
|
| 22 |
# -----------------------------
|