Spaces:
Sleeping
Sleeping
File size: 384 Bytes
d27576c 85e9d26 79ded9c d27576c |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 |
import streamlit as st
from main import save_image
from ultralytics import YOLO
import shutil
model = YOLO('yolov8m.pt')
st.title('Object Detection')
save_image()
if st.button("Detect Object"):
model.predict(source = 'processed_image.jpg', save = True, conf = 0.7)
st.image('runs/detect/predict/processed_image.jpg', use_column_width = 'always')
shutil.rmtree('runs')
|