PHS-BERT / app.py
Chan Lee
M
6973085
raw
history blame
187 Bytes
import ktrain
import gradio as gr
model = ktrain.load_predictor('bert').model
def func(text):
return model.predict(text)
gr.Interface(fn=func, inputs="text", outputs="text").launch()