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)