update

app.py

@@ -106,7 +106,7 @@ def predict(image_path):
 model_selector = gr.Dropdown(
     choices=[(alias, filename) for filename, alias in model_aliases.items()],
     label="Select Model",
-    value="model.pth"
+    value="resnet_model.pth"
 )
 
 load_button = gr.Button("Load Model")
@@ -123,7 +123,6 @@ with gr.Blocks() as demo:
                 examples=[
                     os.path.join("examples", f)
                     for f in os.listdir("examples")
-                    if f.lower().endswith((".jpg", ".jpeg", ".png"))
                 ],
                 inputs=image_input,
                 label="Example Images",