Fix: ensure image is PIL.Image for Hugging Face pipeline

app.py

@@ -1,10 +1,20 @@
 import gradio as gr
 from transformers import pipeline
+from PIL import Image  # ✅ make sure PIL is imported
 
+# Load your model
 model = pipeline("image-classification", model="wellCh4n/tomato-leaf-disease-classification-vit")
 
 def classify(image):
+    if not isinstance(image, Image.Image):
+        image = Image.fromarray(image)  # fallback, just in case
     result = model(image)
-    return result
+    return result[0]['label']  # return top label
 
-gr.Interface(fn=classify, inputs="image", outputs="label").launch()
+gr.Interface(
+    fn=classify,
+    inputs=gr.Image(type="pil"),
+    outputs="label",
+    title="Tomato Leaf Disease Analyzer",
+    description="Upload a tomato leaf image to detect the disease."
+).launch()