Commit
·
0625831
1
Parent(s):
f4fb2f2
no compile
Browse files- utils/color_model.py +2 -1
utils/color_model.py
CHANGED
|
@@ -9,7 +9,8 @@ model_path = hf_hub_download(
|
|
| 9 |
)
|
| 10 |
|
| 11 |
# Load Keras model
|
| 12 |
-
model = tf.keras.models.load_model(model_path)
|
|
|
|
| 13 |
|
| 14 |
def infer_color(vad):
|
| 15 |
input_data = np.array([[
|
|
|
|
| 9 |
)
|
| 10 |
|
| 11 |
# Load Keras model
|
| 12 |
+
model = tf.keras.models.load_model(model_path, compile=False)
|
| 13 |
+
|
| 14 |
|
| 15 |
def infer_color(vad):
|
| 16 |
input_data = np.array([[
|