PHS-BERT / app.py
publichealthsurveillance's picture
Update app.py
73adea4
raw
history blame
218 Bytes
import ktrain
import gradio as gr
predictor = ktrain.load_predictor('bert')
def func(text):
results = predictor.predict(str(text))
return str(results)
gr.Interface(fn=func, inputs="text", outputs="text").launch()