rkonan commited on
Commit
7f8732e
·
1 Parent(s): 58919ac
Files changed (1) hide show
  1. app/model.py +2 -2
app/model.py CHANGED
@@ -41,10 +41,10 @@ class ModelStruct(TypedDict):
41
 
42
 
43
  def load_models() -> list[ModelStruct]:
44
- model1 = tf.keras.models.load_model("model/best_EfficientNetV2M_03_improved_global_augment.keras",compile=False)
45
  model1_for_gradcam=model1
46
 
47
- model2 = tf.keras.models.load_model("model/best_ResNet50V2_04_improved_target_augment.keras",compile=False)
48
  model2_for_gradcam=model2
49
 
50
  return [
 
41
 
42
 
43
  def load_models() -> list[ModelStruct]:
44
+ model1 = tf.keras.models.load_model("model/best_efficientnetv2m_gradcam.keras",compile=False)
45
  model1_for_gradcam=model1
46
 
47
+ model2 = tf.keras.models.load_model("model/best_ResNet50V2_gradcam.keras",compile=False)
48
  model2_for_gradcam=model2
49
 
50
  return [