Update app.py

app.py

@@ -73,7 +73,7 @@ with gr.Blocks(title="MASFNet Object Detection",
     gr.HTML("""<h4 style="color:navy;">4. Then, click "Infer" button to predict object instances. It will take about 10 seconds (yolos-tiny) or 20 seconds (yolos-small).</h4>""")
 
     send_btn = gr.Button("Infer")
-    send_btn.click(fn=infer, inputs=[input_image], outputs=[output_image])
+    send_btn.click(fn=predict, inputs=[input_image], outputs=[output_image])
 
     gr.HTML("""<br/>""")
     gr.HTML("""<h4 style="color:navy;">Reference</h4>""")