Spaces:
Sleeping
Sleeping
Commit
·
963d4f4
1
Parent(s):
81ede04
Update appTest.py
Browse files- appTest.py +1 -1
appTest.py
CHANGED
@@ -71,7 +71,7 @@ def load_model():
|
|
71 |
|
72 |
CNN = Model(inputs=conv_base.input, outputs=output_layer)
|
73 |
# CNN.load_weights("CNN.hdf5")
|
74 |
-
|
75 |
|
76 |
return CNN, model
|
77 |
|
|
|
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 |
|