Update app.py

app.py

@@ -78,11 +78,14 @@ with gr.Blocks() as demo:
     with gr.Row():
         with gr.Column():
             img_input = gr.Image(type="pil", label="Upload an Image")
+            mode_input = gr.Radio(choices=["predict", "dir_predict"], label="Mode", value="predict")
+            crop_input = gr.Checkbox(value=False, label="Crop")
+            count_input = gr.Checkbox(value=True, label="Count")
             submit_btn = gr.Button("Submit")
         with gr.Column():
             output = gr.Image(type="pil", label="Prediction Result")
     
-    submit_btn.click(fn=inference, inputs=[img_input, 'predict', False, True], outputs=output)
+    submit_btn.click(fn=inference, inputs=[img_input, mode_input, crop_input, count_input], outputs=output)
     demo.load(reset_interface, None, [output])
 
     with gr.Row():
@@ -92,3 +95,4 @@ with gr.Blocks() as demo:
         )
 
 demo.launch()
+