Update app.py

app.py

@@ -18,13 +18,13 @@ def normalize_image(img):
 def load_model(model_name):
     # Load the model based on the model_name input
     if model_name == "model1":
-        return tf.keras.models.load_model("Xception.h5")
+        return tf.keras.models.load_model("model/Xception.h5")
     elif model_name == "model2":
-        return tf.keras.models.load_model("InceptionV3.h5")
+        return tf.keras.models.load_model("model/InceptionV3.h5")
     elif model_name == "model3":
-        return tf.keras.models.load_model("InceptionResNetV2.h5")
+        return tf.keras.models.load_model("model/InceptionResNetV2.h5")
     elif model_name == "model4":
-        return tf.keras.models.load_model("DenseNet201.h5")
+        return tf.keras.models.load_model("model/DenseNet201.h5")
     else:
         raise ValueError("Invalid model_name")