Spaces:
Runtime error
Runtime error
| import streamlit as st | |
| from ultralytics import YOLO | |
| from PIL import Image | |
| model= YOLO ("yolov8n-seg.pt") | |
| st.title("image segmentation") | |
| up= st.file_uploader("upload an image", type= ["jpg","png", "jpeg"]) | |
| if st.button("detect"): | |
| image= Image.open(up).convert('RGB') | |
| result= model(image) | |
| output= result[0].plot() | |
| col1, col2= st.columns(2) | |
| with col1: | |
| st.image(image,use_column_width= True) | |
| with col2: | |
| st.image(output,use_column_width= True) | |