Spaces:

Sa-m
/

Brand-Logo-Classification

Runtime error

Sa-m commited on Jun 18

Commit

96d2d57

verified ·

1 Parent(s): 9b52d81

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -11,7 +11,7 @@ from tensorflow_addons.metrics import F1Score
 from keras.utils import custom_object_scope
 with custom_object_scope({'Addons>F1Score': F1Score}):
-    model = tf.keras.models.load_weights('best_model_76.h5')
 def classify_image(inp):

 from keras.utils import custom_object_scope
 with custom_object_scope({'Addons>F1Score': F1Score}):
+    model = tf.keras.models.load_model('best_model.h5')
 def classify_image(inp):