PHS-BERT / app.py
publichealthsurveillance's picture
Update app.py
3d3d5f3
raw
history blame
258 Bytes
import ktrain
import gradio as gr
predictor = ktrain.load_predictor('bert')
def func(text):
results = predictor.predict(text)
results = [int(element) for element in results]
return results
gr.Interface(fn=func, inputs="text", outputs="text").launch()