Update app.py

app.py

@@ -33,19 +33,26 @@ if __name__ == "__main__":
     def inference(img, confidence):
         img = np.array(img)
         _, results = vis_demo.run_on_image(img, confidence)
-        results = Image.fromarray(results.get_image())
+        results = Image.fromarray(results.get_image()[:, :, ::-1])
         return results 
     
     demo = gr.Interface(
         fn=inference,
         inputs=[
             gr.Image(type="pil", image_mode="RGB"),
-            gr.Number(precision=2, minimum=0.0, maximum=1.0, value=0.5)
+            # gr.Number(precision=2, minimum=0.0, maximum=1.0, value=0.5)
+            gr.Slider(minimum=0.0, maximum=1.0, value=0.5, step=0.05)
         ],
         outputs="image",
         examples=[
             ["MrDETR/assets/000000014226.jpg", 0.5],
-            ["MrDETR/assets/000000028449.jpg", 0.5]
+            ["MrDETR/assets/000000028449.jpg", 0.3],
+            ["MrDETR/assets/000000070048.jpg", 0.5],
+            ["MrDETR/assets/000000218997.jpg", 0.5],
+            ["MrDETR/assets/000000279774.jpg", 0.5],
+            ["MrDETR/assets/000000434459.jpg", 0.5],
+            ["MrDETR/assets/000000448448.jpg", 0.5],
+            ["MrDETR/assets/000000560474.jpg", 0.5],
         ]
     )
     demo.launch()