File size: 739 Bytes
b454b86 3de3309 b454b86 3de3309 b454b86 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 |
import glob
import gradio as gr
from inference import *
from PIL import Image
def gradio_app(image_path):
"""A function that send the file to the inference pipeline, and filters
some predictions before outputting to gradio interface."""
predictions = run_inference(image_path)
out_img = Image.fromarray(predictions.render()[0])
return out_img
Title = "Marine Life Identification"
description = (
""
)
examples = glob.glob("images/*.png")
gr.Interface(gradio_app,
inputs=[gr.inputs.Image(type="filepath")],
outputs=gr.outputs.Image(type="pil"),
enable_queue=True,
title=Title,
description=description,
examples=examples).launch() |