Update app.py

app.py

@@ -115,10 +115,10 @@ demo = gr.Interface(
     fn=yolo_predict,
     inputs=[
         gr.Image(shape=(config.IMAGE_SIZE,config.IMAGE_SIZE), label="Input Image"),
-        gr.Slider(0, 1, value=0.8, step=0.05, label="IOU Threshold"),
-        gr.Slider(0, 1, value=0.8, step=0.05, label="Threshold")
+        gr.Slider(0, 1, value=0.75, step=0.05, label="IOU Threshold"),
+        gr.Slider(0, 1, value=0.75, step=0.05, label="Threshold")
     ],
-    outputs=[gr.Gallery(label="Output Image", columns=1, rows=1)],
+    outputs=gr.Image(label="Prediction"),
     examples=[
        ["examples/000001.jpg", 0.75, 0.75],
        ["examples/000002.jpg", 0.75, 0.75],