monology's picture
Update app.py
45c06f4
raw
history blame
351 Bytes
from transformers import pipeline
from PIL import Image
def predict(image):
pipe = pipeline("image-classification", model="dennisjooo/Birds-Classifier-EfficientNetB2")
predictions = pipe(image)
return {p["label"].title(): p["score"] for p in predictions}
import gradio as gr
gr.Interface(predict, gr.Image(type="pil"), "label").launch()