Update app.py

app.py

@@ -27,8 +27,11 @@ if uploaded_file:
     image = Image.open(uploaded_file)
     image_np = np.array(image)
 
+    # Display "Running inference..." in red
+    placeholder = st.empty()
+    placeholder.markdown('<h3 style="color: red;">Running inference...</h3>', unsafe_allow_html=True)
+
     # Run inference
-    st.write("Running inference...")
     results = model.predict(source=image_np, conf=0.5)
 
     # Annotate image
@@ -39,11 +42,5 @@ if uploaded_file:
     # Convert annotated image for Streamlit
     if annotated_image is not None:
         annotated_image_rgb = cv2.cvtColor(annotated_image, cv2.COLOR_BGR2RGB)
-        st.image(annotated_image_rgb, caption="Prediction Results", use_column_width=True)
-
-    # Display bounding boxes and confidence scores
-    st.write("Detection Results:")
-    for result in results:
-        for box in result.boxes.data.cpu().numpy():
-            x1, y1, x2, y2, conf, cls = box
-            st.write(f"Class: {int(cls)}, Confidence: {conf:.2f}, Box: [{x1:.0f}, {y1:.0f}, {x2:.0f}, {y2:.0f}]")
+        placeholder.empty()  # Remove the "Running inference..." message
+        st.image(annotated_image_rgb, caption="Prediction Results", use_container_width=True)