import gradio as gr

from transformers import pipeline



pipeline = pipeline(task="image-classification", model="RGurung/Animal_model")


def predict(input_img):

  predictions = pipeline(input_img)

  return input_img, {p["label"]: p["score"] for p in predictions}


gradio_app = gr.Interface(

    predict,

    inputs=gr.Image(label = "Select Image", sources=['upload', 'webcam'], type='pil'),

    outputs=[gr.Image(label = "Processed Image"), gr.Label(label = "Result", num_top_classes=2)],

    title="Clownfish or Panda?",

)


if __name__ == "__main__":

    gradio_app.launch()