Spaces:
Running
Running
Commit
·
7958075
1
Parent(s):
4db3f74
Update app.py
Browse files
app.py
CHANGED
|
@@ -7,9 +7,6 @@ from utils import *
|
|
| 7 |
|
| 8 |
# Initialize and build the model first
|
| 9 |
model = model_arc()
|
| 10 |
-
dummy_input = tf.zeros((1, 256, 256, 3))
|
| 11 |
-
model.build(input_shape=dummy_input.shape)
|
| 12 |
-
model(dummy_input)
|
| 13 |
model.load_weights("classify_model.h5")
|
| 14 |
|
| 15 |
labels = gen_labels()
|
|
|
|
| 7 |
|
| 8 |
# Initialize and build the model first
|
| 9 |
model = model_arc()
|
|
|
|
|
|
|
|
|
|
| 10 |
model.load_weights("classify_model.h5")
|
| 11 |
|
| 12 |
labels = gen_labels()
|