Spaces:
Sleeping
Sleeping
Commit
•
81ede04
1
Parent(s):
6d35101
Update appTest.py
Browse files- appTest.py +2 -1
appTest.py
CHANGED
|
@@ -70,7 +70,8 @@ def load_model():
|
|
| 70 |
output_layer = Dense(5, activation='softmax')(top_model)
|
| 71 |
|
| 72 |
CNN = Model(inputs=conv_base.input, outputs=output_layer)
|
| 73 |
-
CNN.load_weights("CNN.hdf5")
|
|
|
|
| 74 |
|
| 75 |
return CNN, model
|
| 76 |
|
|
|
|
| 70 |
output_layer = Dense(5, activation='softmax')(top_model)
|
| 71 |
|
| 72 |
CNN = Model(inputs=conv_base.input, outputs=output_layer)
|
| 73 |
+
# CNN.load_weights("CNN.hdf5")
|
| 74 |
+
CNN.load_weights("tl_model_v1.weights.best.hdf5")
|
| 75 |
|
| 76 |
return CNN, model
|
| 77 |
|