Sa-m commited on
Commit
f9ead50
·
verified ·
1 Parent(s): 1be3e87

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -3
app.py CHANGED
@@ -11,10 +11,8 @@ 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.h5')
15
 
16
- # Loading trained model
17
- # model = tf.keras.models.load_model('best_model2.h5')
18
 
19
  def classify_image(inp):
20
  np.random.seed(143)
 
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):
18
  np.random.seed(143)