app.py

import gradio as gr
from transformers import pipeline

# Carregar o modelo de classificação de feijão
classifier = pipeline("image-classification", model="mestrevh/computer-vision-beans")

# Função para classificar a imagem
def classify_bean(image):
    result = classifier(image)
    return result[0]['label'], result[0]['score']

# Criar a interface com Gradio
iface = gr.Interface(
    fn=classify_bean,  # Função que será chamada
    inputs=gr.inputs.Image(type="pil"),  # Tipo de entrada (imagem)
    outputs=["text", "json"],  # O que será exibido após a classificação
    live=True,  # Atualizar automaticamente
    title="Classificação de Feijões",  # Título do aplicativo
    description="Faça o upload de uma imagem de feijão e descubra a classificação!",  # Descrição
)

# Rodar o aplicativo
iface.launch()