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('phsbert')

def bert(text):
	results = predictor_bert.predict(str(text))
	return str(results)
	
def mentalbert(text):
	results = predictor_mental.predict(str(text))
	return str(results)
	
def phsbert(text):
	results = predictor_phs.predict(str(text))
	return str(results)

bert_io = gr.Interface(fn=bert, inputs="text", outputs="text")
mental_io = gr.Interface(fn=mentalbert, inputs="text", outputs="text")
# phs_io = gr.Interface(fn=phsbert, inputs="text", outputs="text")

# Parallel(bert_io, mental_io, phs_io).launch()
Parallel(bert_io, mental_io).launch()
# bert_io.launch()
# mental_io.launch()
# phs_io.launch()