Spaces:
Paused
Paused
ajout
Browse files- 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/
|
| 45 |
model1_for_gradcam=model1
|
| 46 |
|
| 47 |
-
model2 = tf.keras.models.load_model("model/
|
| 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 [
|