Update app.py

app.py

@@ -22,6 +22,7 @@ vit_label_treatment = {
     "Potato___Late_blight": "Use certified seed tubers; fungicide sprays when conditions favor disease.",
     "Rice___Leaf_blight": "Use resistant rice varieties, maintain field hygiene.",
     "Rice___Brown_spot": "For brown spot of rice, employ integrated disease management practices. Use disease-resistant rice varieties and ensure balanced fertilization to prevent excessive nitrogen application. Maintain optimal irrigation to avoid water stress, and at the first signs of infection, apply fungicides such as carbendazim or validamycin following label recommendations. Remove and destroy infected plant debris to reduce inoculum, and consider crop rotation to mitigate disease recurrence.",
+    "Rice___Brown_Spot": "For brown spot of rice, employ integrated disease management practices. Use disease-resistant rice varieties and ensure balanced fertilization to prevent excessive nitrogen application. Maintain optimal irrigation to avoid water stress, and at the first signs of infection, apply fungicides such as carbendazim or validamycin following label recommendations. Remove and destroy infected plant debris to reduce inoculum, and consider crop rotation to mitigate disease recurrence.",
     "Wheat___Leaf_rust": "Plant resistant wheat varieties, apply foliar fungicides if severe.",
     # Fallback
     "Unknown": "No specific treatment available."
@@ -187,8 +188,8 @@ def classify_and_visualize_keras(image):
 def main_model_selector(model_choice, image):
     """
     Dispatch function based on user choice of model:
-      - 'Vit-model (Corn, Potato, Rice, Wheat)' -> use classify_image_vit
-      - 'Keras-model (Apple, Blueberry, Cherry, etc.)' -> use classify_and_visualize_keras
+      - 'ViT (Corn, Potato, Rice, Wheat)' -> use classify_image_vit
+      - 'Keras (Apple, Blueberry, Cherry, etc.)' -> use classify_and_visualize_keras
     """
     if image is None:
         return "No image provided.", None, None, None
@@ -256,4 +257,4 @@ with gr.Blocks() as demo:
         inputs=[model_choice, inp_image]
     )
 
-demo.launch(share=True)
+demo.launch(share=True)