semantic-seg / app.py
sofmi's picture
app.py file (in progress)
04e445b
raw
history blame
395 Bytes
from huggingface_hub import from_pretrained_keras
import gradio as gr
import tensorflow as tf
import numpy as np
import os
# load model
model = tf.keras.models.load_model('keras-io/semantic-segmentation')
# gradio components
inputs = gr.inputs.Image()
outputs = gr.outputs.Image()
# inference fn
def predict(image_input):
pass
gr.Interface(predict, inputs = inputs, outputs = outputs)