Spaces:

DFisch
/

Image-Manipulation-Detection

Running

David Fischinger commited on Nov 8, 2024

Commit

1530a3e

1 Parent(s): 6c58fe6

fix for loading model

Files changed (1) hide show

app.py CHANGED Viewed

@@ -27,8 +27,8 @@ def load_models():
     model_path1 = "IMVIP_Supplementary_Material/models/model1/"
     model_path2 = "IMVIP_Supplementary_Material/models/model2/"
-    model_M1 = tf.keras.models.load_model("IMVIP_Supplementary_Material/models/model1/model1.h5")
-    model_M2 = tf.keras.models.load_model("IMVIP_Supplementary_Material/models/model2/model2.h5")
     return model_M1, model_M2

     model_path1 = "IMVIP_Supplementary_Material/models/model1/"
     model_path2 = "IMVIP_Supplementary_Material/models/model2/"
+    model_M1 = tf.keras.models.load_model("IMVIP_Supplementary_Material/models/model1/")
+    model_M2 = tf.keras.models.load_model("IMVIP_Supplementary_Material/models/model2/")
     return model_M1, model_M2