import gradio as gr

title = "Digit Classifier - By Teeny-Tiny Castle 🏰"

head = (
  "<center>"
  "<img src='https://upload.wikimedia.org/wikipedia/commons/2/27/MnistExamples.png' width=400>"
  "This model was trained to classify numbers (from 0 to 9). To test it, write your number in the space provided."
  "</center>"
)


ref = (
"<center>"
"Return to the <a href='https://github.com/Nkluge-correa/teeny-tiny_castle)'>castle</a>."
"</center>")

# create interface
demo = gr.Interface(fn=predict,
             inputs="sketchpad",
             outputs=gr.Label(num_top_classes=3),
             allow_flagging="never",
             title=title,
             description=head,
             article=ref)

# launch interface
demo.launch(inline=False)