Update app.py

app.py

@@ -6,7 +6,7 @@ import gradio as gr
 import os
 
 # Load model
-checkpoint_path = "edgilr/model_checkpoint.pth"
+checkpoint_path = "edgilr/fasterRCNNKangaroo.pth"
 checkpoint_and_model = model_from_checkpoint(checkpoint_path)
 model = checkpoint_and_model["model"]
 model_type = checkpoint_and_model["model_type"]
@@ -18,9 +18,9 @@ valid_tfms = tfms.A.Adapter([*tfms.A.resize_and_pad(img_size), tfms.A.Normalize(
 
 # Populate examples in Gradio interface
 examples = [
-    ['1.jpg'],
-    ['2.jpg'],
-    ['3.jpg']
+    ['00004.jpg'],
+    ['00083.jpg'],
+    ['00119.jpg']
 ]
 
 def show_preds(input_image):
@@ -38,9 +38,9 @@ def show_preds(input_image):
 gr_interface = gr.Interface(
     fn=show_preds,
     inputs=["image"],
-    outputs=[gr.outputs.Image(type="pil", label="RetinaNet Inference")],
-    title="Fridge Object Detector",
-    description="A VFNet model that detects common objects found in fridge. Upload an image or click an example image below to use.",
+    outputs=[gr.outputs.Image(type="pil", label="FasterRCNN Inference")],
+    title="Kangaroo Object Detector",
+    description="",
     examples=examples,
 )
 gr_interface.launch(inline=False, share=False, debug=True)