import gradio as gr
from inference import predict

demo = gr.Interface(fn=predict, inputs='text', outputs='text')
demo.launch()