Update app.py

app.py

@@ -9,20 +9,20 @@ feature_extractor = ViTFeatureExtractor.from_pretrained("iamomtiwari/VITPEST")
 
 # Define class labels and treatment advice
 class_labels = {
-    "Corn___Common_Rust": "Apply fungicides as soon as symptoms are noticed. Practice crop rotation and remove infected plants.",
-    "Corn___Gray_Leaf_Spot": "Rotate crops to non-host plants, apply resistant varieties, and use fungicides as needed.",
-    "Corn___Healthy": "Continue good agricultural practices: ensure proper irrigation, nutrient supply, and monitor for pests.",
-    "Corn___Northern_Leaf_Blight": "Remove and destroy infected plant debris, apply fungicides, and rotate crops.",
-    "Rice___Brown_Spot": "Use resistant varieties, improve field drainage, and apply fungicides if necessary.",
-    "Rice___Healthy": "Maintain proper irrigation, fertilization, and pest control measures.",
-    "Rice___Leaf_Blast": "Use resistant varieties, apply fungicides during high-risk periods, and practice good field management.",
-    "Rice___Neck_Blast": "Plant resistant varieties, improve nutrient management, and apply fungicides if symptoms appear.",
-    "Wheat___Brown_Rust": "Apply fungicides and practice crop rotation with non-host crops.",
-    "Wheat___Healthy": "Continue with good management practices, including proper fertilization and weed control.",
-    "Wheat___Yellow_Rust": "Use resistant varieties, apply fungicides, and rotate crops.",
-    "Sugarcane__Red_Rot": "Plant resistant varieties and ensure good drainage.",
-    "Sugarcane__Healthy": "Maintain healthy soil conditions and proper irrigation.",
-    "Sugarcane__Bacterial Blight": "Use disease-free planting material, practice crop rotation, and destroy infected plants."
+    0:"Corn___Common_Rust": "Apply fungicides as soon as symptoms are noticed. Practice crop rotation and remove infected plants.",
+    1:"Corn___Gray_Leaf_Spot": "Rotate crops to non-host plants, apply resistant varieties, and use fungicides as needed.",
+    2:"Corn___Healthy": "Continue good agricultural practices: ensure proper irrigation, nutrient supply, and monitor for pests.",
+    3:"Corn___Northern_Leaf_Blight": "Remove and destroy infected plant debris, apply fungicides, and rotate crops.",
+    4:"Rice___Brown_Spot": "Use resistant varieties, improve field drainage, and apply fungicides if necessary.",
+    5:"Rice___Healthy": "Maintain proper irrigation, fertilization, and pest control measures.",
+    6:"Rice___Leaf_Blast": "Use resistant varieties, apply fungicides during high-risk periods, and practice good field management.",
+    7:"Rice___Neck_Blast": "Plant resistant varieties, improve nutrient management, and apply fungicides if symptoms appear.",
+    8:"Wheat___Brown_Rust": "Apply fungicides and practice crop rotation with non-host crops.",
+    9:"Wheat___Healthy": "Continue with good management practices, including proper fertilization and weed control.",
+    10:"Wheat___Yellow_Rust": "Use resistant varieties, apply fungicides, and rotate crops.",
+    11:"Sugarcane__Red_Rot": "Plant resistant varieties and ensure good drainage.",
+    12:"Sugarcane__Healthy": "Maintain healthy soil conditions and proper irrigation.",
+    13:"Sugarcane__Bacterial Blight": "Use disease-free planting material, practice crop rotation, and destroy infected plants."
 }
 
 # Mapping label indices to class labels