Update app.py

app.py

@@ -7,9 +7,14 @@ import os
 HEIGHT, WIDTH = 224, 224
 NUM_CLASSES = 6
 LABELS = ["Burger King", "KFC", "McDonalds", "Other", "Starbucks", "Subway"]
+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_model('best_model2.h5')
 
 # Loading trained model
-model = tf.keras.models.load_model('best_model2.h5')
+# model = tf.keras.models.load_model('best_model2.h5')
 
 def classify_image(inp):
     np.random.seed(143)