Mariam-recon / app.py
Docfile's picture
Create app.py
8a1f264 verified
raw
history blame
493 Bytes
import gradio as gr
import os
from ultralytics import YOLO
model = YOLO("yolov8n.pt") # load a pretrained model (recommended for training)
# Chargez l'image
e =""
# Fonction pour générer le contenu
def generate_content(pro,image):
global e
results = model(image)
e = results
return e
markdown = r"""
e
""".format(e)
# Interface Gradio
iface = gr.Interface(fn=generate_content, inputs=[gr.Textbox(),gr.Image(type='pil')], outputs= gr.Image(type='pil'))
iface.launch()