Spaces:
Sleeping
Sleeping
| import cv2 | |
| import numpy as np | |
| import gradio as gr | |
| from ultralytics import YOLO | |
| def predict(path:str): | |
| model = YOLO("yolo11s.yaml") | |
| model = YOLO("constellations_best.pt") | |
| imagen = cv2.imread(path) | |
| results = model.predict(source=path) | |
| for r in results: | |
| return r.plot() | |
| gr.Interface(fn=predict, | |
| inputs=gr.components.Image(type="filepath", label="Input"), | |
| outputs=gr.components.Image(type="numpy", label="Output")).launch(debug=False) |