Hugging Face's logo

Spaces:
Sa-m
/
Brand-Logo-Classification
Running

App Files Community
Sa-m commited on
Commit
9b52d81
·
verified ·
1 Parent(s): f9ead50

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -11,7 +11,7 @@ from tensorflow_addons.metrics import F1Score
11
  from keras.utils import custom_object_scope
12
 
13
  with custom_object_scope({'Addons>F1Score': F1Score}):
14
- model = tf.keras.models.load_model('best_model_76.h5')
15
 
16
 
17
  def classify_image(inp):
 
11
  from keras.utils import custom_object_scope
12
 
13
  with custom_object_scope({'Addons>F1Score': F1Score}):
14
+ model = tf.keras.models.load_weights('best_model_76.h5')
15
 
16
 
17
  def classify_image(inp):