App install correct

app.py

@@ -1,8 +1,10 @@
 import gradio as gr
-from transformers import pipeline
+from transformers import AutoModelForImageClassification, AutoFeatureExtractor, pipeline
 
-# Carregar o modelo diretamente do Hugging Face Hub
-classifier = pipeline("image-classification", model="mestrevh/computer-vision-beans")
+model = AutoModelForImageClassification.from_pretrained("mestrevh/computer-vision-beans", use_safetensors=True)
+feature_extractor = AutoFeatureExtractor.from_pretrained("mestrevh/computer-vision-beans")
+
+classifier = pipeline("image-classification", model=model, feature_extractor=feature_extractor)
 
 # Função de classificação
 def predict_image(image):