Spaces:
Sleeping
Sleeping
import json | |
import gradio | |
import numpy as np | |
from model.model import Model | |
def main(list_encoded_image: str): | |
image_array = np.array(json.loads(list_encoded_image), dtype=np.uint8) | |
image = Model().detect_object(data=image_array) | |
list_encoded_image = np.frombuffer(image, dtype=np.uint8).tolist() | |
return json.dumps(list_encoded_image) | |
gradio_interface = gradio.Interface(fn=main, inputs="text", outputs="text") | |
gradio_interface.launch() |