Update app.py

app.py

@@ -8,8 +8,8 @@ class_names = ['AnnualCrop', 'Forest', 'HerbaceousVegetation', 'Highway', 'Indus
 
 # Function to load the selected model
 def load_selected_model(model_name):
-    if model_name == "ResNet50_Model.keras":
-        return load_model('ResNet50_Model.keras')
+    if model_name == "ResNet50_model.h5":
+        return load_model('ResNet50_model.h5')
     elif model_name == "CNN_model.h5":
         return load_model('CNN_model.h5')
     else:
@@ -46,7 +46,7 @@ with gr.Blocks() as demo:
 
     with gr.Row():
         with gr.Column():
-            model_selector = gr.Dropdown(choices=["ResNet50_Model.keras", "CNN_model.h5"], label="Select Model")
+            model_selector = gr.Dropdown(choices=["ResNet50_model.h5", "CNN_model.h5"], label="Select Model")
             image_input = gr.Image(type="numpy", label="Upload Image")
             predict_button = gr.Button("Predict")