Update app.py

app.py

@@ -37,8 +37,6 @@ def classify_image_vit(image):
 
     # Predicted label
     predicted_label = vit_model.config.id2label.get(predicted_class_idx, "Unknown")
-    # Example: If your id2label from HF is something like "corn diseased" or "rice healthy",
-    # match it to the dictionary key for treatments (above). For demonstration:
     treatment_text = vit_label_treatment.get(predicted_label, "No specific treatment available.")
     return predicted_label, treatment_text