modified the app.py

app.py

@@ -1,9 +1,9 @@
 import gradio as gr
-from transformers import AutoModelForImageClassification, AutoImageProcessor, pipeline
+from transformers import AutoModelForImageClassification, AutoFeatureExtractor, pipeline
 
-# Carregar o modelo e o processador de imagens
+# Carregar o modelo e o processador de imagens (AutoFeatureExtractor ao invés de AutoImageProcessor)
 model = AutoModelForImageClassification.from_pretrained("mestrevh/computer-vision-beans", use_safetensors=True)
-image_processor = AutoImageProcessor.from_pretrained("mestrevh/computer-vision-beans")
+image_processor = AutoFeatureExtractor.from_pretrained("mestrevh/computer-vision-beans")
 
 # Criar o pipeline
 classifier = pipeline("image-classification", model=model, feature_extractor=image_processor)
@@ -11,6 +11,7 @@ classifier = pipeline("image-classification", model=model, feature_extractor=ima
 # Função de classificação
 def predict_image(image):
     # A saída do classifier é uma lista de dicionários, pegar o label e a confiança
+    print("test")
     result = classifier(image)
     label = result[0]['label']
     confidence = result[0]['score']