File size: 527 Bytes
6391a72 70a7569 af2fb90 1fe8523 6ef4aaa af2fb90 6dae3a6 45af5d2 5dfbcd0 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 |
import gradio as gr
from PIL import Image
from fastapi import FastAPI
from process import Segment
import numpy as np
app = FastAPI()
def process_image(input_image):
if not isinstance(input_image, Image.Image):
input_image = Image.fromarray(input_image.astype('uint8'), 'RGB')
output_image = Segment(input_image, [0, 0, 0])
return output_image
iface = gr.Interface(
fn=process_image,
inputs=gr.components.Image(type="numpy"),
outputs=gr.components.Image(type="numpy")
)
iface.launch()
|