legalis / app.py
LennardZuendorf's picture
updating interface to integrate both models
9142a0b
import gradio as gr
tts_examples = [
"I love learning machine learning",
"How do you do?",
]
scikit_demo = gr.load(
"LennardZuendorf/legalis-scikit",
src="models",
inputs=gr.Textbox(lines=5, max_lines=6, label="Input Text"),
title="scikit-learn"
)
bert_demo = gr.load(
"LennardZuendorf/legalis-bert",
src="models",
inputs=gr.Textbox(lines=5, max_lines=6, label="Input Text"),
title="bert"
)
demo = gr.TabbedInterface([scikit_demo, bert_demo], ["BERT Classification", "scikit-learn Classification"])
if __name__ == "__main__":
demo.launch()