import gradio as gr

gr.load(
    name="JanSt/albert-base-v2_mbti-classification",
    src="models",
    title="Toxic comment detection",
    description="Say if the prompt is toxic or not.",
    article="Check out the [model repo](https://huggingface.co/tillschwoerer/roberta-base-finetuned-toxic-comments-detection) that this demo is based off of.",
    examples=[["""Let's focus on how we can use our talents and resources to make a positive difference -- I believe in everyone's potential to make a meaningful contribution."""], 
              ["""Let's step back and consider the broad perspective — I'm sure we can identify the best path forward if we focus on the bigger picture."""], 
              ["""Let's have some fun and let our imaginations run wild — I'm sure we can come up with something enjoyable if we tap into our creative energy."""]],
).launch()