Update app.py

app.py

@@ -15,18 +15,15 @@ def predict_with_yolo(image):
     # Run inference on the image using the YOLO model
     results = st.session_state.yolo_model(image)
     
-    # Extract predictions
-    predictions = []
+    # Print the classification results
     if results:
+        # YOLOv8 classification results contain label and confidence for the whole image
         for result in results:
-            for box in result.boxes:
-                class_name = result.names[box.label]
-                confidence = box.conf.item()  # Convert tensor to a Python float
-                predictions.append({
-                    "Class": class_name,
-                    "Confidence": confidence
-                })
-    return predictions
+            # print(f"Class: {result.names[result.probs.top1]}, Confidence: {result.probs.top1conf:.2f}")
+            return {"Class":result.names[result.probs.top1],"Confidence":result.probs.top1conf:.2f}
+    else:
+        print("No classification results found.")
+    
 
 # Streamlit app UI
 st.title("Clothing Detection with YOLO")
@@ -46,9 +43,7 @@ if url:
 
             # Display predictions
             if predictions:
-                st.write("Predictions:")
-                for pred in predictions:
-                    st.write(f"Class: {pred['Class']}, Confidence: {pred['Confidence']:.2f}")
+                st.write("Predictions:",predictions)
             else:
                 st.write("No objects detected.")