Spaces:
Runtime error
Runtime error
# import gradio as gr | |
# def greet(name): | |
# return "Hello " + name + "!!" | |
# demo = gr.Interface(fn=greet, inputs="text", outputs="text") | |
# demo.launch() | |
import gradio as gr | |
from ultralytics import YOLO | |
from PIL import Image | |
#Load model | |
model = YOLO('yolov8n.pt') | |
def image_display(input_image): | |
# Return the input image as the output | |
model = YOLO('yolov8n.pt') | |
results = model(input_image) | |
for r in results: | |
im_array = r.plot() # plot a BGR numpy array of predictions | |
im = Image.fromarray(im_array[..., ::-1]) # RGB PIL image | |
return im_array | |
input_component = gr.inputs.Image() | |
output_component = gr.outputs.Image() | |
# Create the Gradio interface | |
gr.Interface( | |
fn=image_display, | |
inputs=input_component, | |
outputs=output_component, | |
title="Image Display App", | |
description="Upload an image and see it displayed.", | |
theme="compact" | |
).launch(share=True) | |
# demo = gr.Interface( | |
# fn=greet, | |
# inputs='textbox', | |
# outputs='textbox', | |
# ) | |
# demo.launch() | |