Upload 3 files

app.py

@@ -0,0 +1,33 @@
+import streamlit as st
+from PIL import Image
+from ultralytics import YOLO
+
+st.title("Object Detection with YOLO")
+
+model = YOLO('best.pt')
+
+def detect_objects(image):
+    result = model(image)
+    return result
+
+
+uploaded_file = st.file_uploader("Choose an image...", type=["jpg", "png", "jpeg"])
+
+if uploaded_file is not None:
+    image = Image.open(uploaded_file)
+    st.image(image, caption='Uploaded Image.', use_column_width=True)
+
+    if st.button('Detect Objects'):
+        image.save('uploaded_image.jpg')
+        
+        result = detect_objects('uploaded_image.jpg')
+        
+        boxes = result[0].boxes
+        masks = result[0].masks
+        keypoints = result[0].keypoints
+        probs = result[0].probs
+        
+        st.write("Number of objects detected:", len(boxes))
+        result[0].save(filename='result.jpg')
+        
+        st.image('result.jpg', use_column_width=True)

best.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:2631a282bc9fe4bf10af87f1ea6a89b2c15f756e22a8c35f24a3e2303243caa6
+size 6223705

requirements.txt

@@ -0,0 +1,3 @@
+torch
+ultralytics
+streamlit