enable api

app.py

@@ -2,17 +2,14 @@ import gradio as gr
 from transformers import pipeline
 from PIL import Image
 
-# Load your model
 model = pipeline("image-classification", model="wellCh4n/tomato-leaf-disease-classification-vit")
 
-# Define the prediction function
 def classify(image):
     if not isinstance(image, Image.Image):
         image = Image.fromarray(image)
     result = model(image)
-    return result[0]['label']  # You can also return full result if you want scores
+    return result[0]['label']
 
-# Define the interface
 app = gr.Interface(
     fn=classify,
     inputs=gr.Image(type="pil"),
@@ -21,5 +18,5 @@ app = gr.Interface(
     description="Upload a tomato leaf image to detect the disease."
 )
 
-# 🚀 Enable both UI and API
+# 🚀 This is important
 app.launch(show_api=True)