Spaces:
Runtime error
Runtime error
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() |