Update app.py

app.py

@@ -1,18 +1,18 @@
 import gradio as gr
 import requests
-from transformers import AutoModelForImageClassification, AutoTokenizer
+from transformers import AutoModelForImageClassification, AutoFeatureExtractor
 from PIL import Image
 import torch
 
-# Menginisialisasi model dan tokenizer dari Hugging Face
-model_name = "ahmadalfian/fruits_vegetables_classifier"  # Model yang kamu sebutkan
+# Menginisialisasi model dan feature extractor dari Hugging Face
+model_name = "ahmadalfian/fruits_vegetables_classifier"
 model = AutoModelForImageClassification.from_pretrained(model_name)
-tokenizer = AutoTokenizer.from_pretrained(model_name)
+feature_extractor = AutoFeatureExtractor.from_pretrained(model_name)
 
 # Fungsi untuk memprediksi kelas
 def predict(image):
-    image = image.convert("RGB")
-    inputs = tokenizer(image, return_tensors="pt")
+    # Preprocess gambar menggunakan feature extractor
+    inputs = feature_extractor(images=image, return_tensors="pt")
     with torch.no_grad():
         outputs = model(**inputs)
     predictions = outputs.logits.argmax(dim=1)
@@ -98,5 +98,4 @@ iface = gr.Interface(
     description="Upload an image of a fruit or vegetable to classify and get its nutritional information."
 )
 
-
 iface.launch()