Test / app.py
JosephH's picture
Update app.py
da509c0
raw
history blame contribute delete
241 Bytes
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()