Update app.py

app.py

@@ -15,17 +15,17 @@ if uploaded_file is not None:
     image = Image.open(uploaded_file)
 
     # Inference
-    results = model.predict(image)
+    results = model(image)
 
     # Display the image with bounding boxes
     st.image(image, channels="RGB", caption="Object Detection Result", use_column_width=True)
 
     # Display probability and class for each box
-    for det in results.boxes.xyxy[0]:
-        label = int(det[5])
-        score = float(det[4])
-        box = det[:4].tolist()
+    for det in results:
+        box = det.boxes.xyxy
+        cls = det.boxes.cls
+        score = det.boxes.conf
 
         # Draw box on the image
-        st.image(model.plot([image], boxes=[box], labels=[label], confidences=[score])[0],
+        st.image(model.plot([image], boxes=[box], labels=[cls], confidences=[score])[0],
                  channels="RGB", caption=f"Class {label} - Confidence: {score:.2f}", use_column_width=True)