Update app.py

fixed rgb format

app.py

@@ -79,10 +79,9 @@ def yolo_inference_tool():
 
         # Annotated image
         try:
-            annotated_img_bgr = r.plot(conf=True, boxes=True, labels=True)
-            annotated_img_rgb = Image.fromarray(annotated_img_bgr[..., ::-1])
+            annotated_img_pil = r.plot(conf=True, boxes=True, labels=True, pil=True)
             st.subheader("Annotated Image")
-            st.image(annotated_img_rgb, caption="Inference Output", use_container_width=True)
+            st.image(annotated_img_pil, caption="Inference Output", use_container_width=True)
         except Exception as e:
             st.error(f"Error generating annotated image: {e}")