Spaces:
Sleeping
Sleeping
import os | |
import gradio as gr | |
from transformers import pipeline | |
od_pipe = pipeline("object-detection", "facebook/detr-resnet-50") | |
def get_pipeline_prediction(pil_image): | |
# first get the pipeline output given the pil image | |
pipeline_output = od_pipe(pil_image) | |
# Then Process the image using the pipeline output | |
processed_image = render_results_in_image(pil_image, | |
pipeline_output) | |
return processed_image | |
demo = gr.Interface( | |
fn=get_pipeline_prediction, | |
inputs=gr.Image(label="Input image", | |
type="pil"), | |
outputs=gr.Image(label="Output image with predicted instances", | |
type="pil") | |
) | |
demo.launch |