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