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