PHS-BERT / app.py
publichealthsurveillance's picture
Update app.py
a8f674d
raw
history blame
782 Bytes
import ktrain
import gradio as gr
from gradio.mix import Parallel
predictor_bert = ktrain.load_predictor('bert')
predictor_mental = ktrain.load_predictor('mentalbert')
predictor_phs = ktrain.load_predictor('phs-bert')
def func_bert(text):
results = predictor_bert.predict(str(text))
return str(results)
def func_mental(text):
results = predictor_mental.predict(str(text))
return str(results)
def func_phs(text):
results = predictor_phs.predict(str(text))
return str(results)
# bert_io = gr.Interface(fn=func_bert, inputs="text", outputs="text")
# mental_io = gr.Interface(fn=func_mental, inputs="text", outputs="text")
phs_io = gr.Interface(fn=func_phs, inputs="text", outputs="text")
# Parallel(bert_io, mental_io, phs_io).launch()
# bert_io.launch()
phs_io.launch()