import gradio as gr
from model import GPT2PPLV2 as GPT2PPL

# initialize the model
model = GPT2PPL()


def greet(sentence):
    return model(sentence, 512, "v1.1")


demo = gr.Interface(fn=greet, inputs="text", outputs="text")

demo.launch()