# import required libraries
from transformers import pipeline
import gradio as gr
import timm

# build gradio interface
model = pipeline("image-classification")
examples = ["birdA.jpg", "birdB.jpg", "birdC.jpg"]
gr.Interface.from_pipeline(model,
                           examples = examples,
                           theme = gr.themes.Soft(),
                           title = "TellMe-AI",
                           css = ".gradio-container {background: #0052D4;background: -webkit-linear-gradient(to top, #6FB1FC, #4364F7, #0052D4);background: linear-gradient(to top, #6FB1FC, #4364F7, #0052D4);}").launch()