Update app.py

app.py

@@ -72,7 +72,7 @@ def load_detectron_model(fruit_name):
 # Labels
 label_map_name = {
     0: "Banana", 1: "Cucumber", 2: "Grape", 3: "Kaki", 4: "Papaya",
-    5: "Peach", 6: "Pear", 7: "Pepper", 8: "Strawberry", 9: "Watermelon",
+    5: "Peach", 6: "Pear", 7: "Peeper", 8: "Strawberry", 9: "Watermelon",
     10: "tomato"
 }
 label_map_quality = {0: "Good", 1: "Mild", 2: "Rotten"}
@@ -110,7 +110,7 @@ def main():
             st.write(f"Fruits Type Detection:  {predicted_name}")
             st.write(f"Fruits Quality Classification:  {predicted_quality}")
 
-            if predicted_name.lower() in ["kaki", "tomato", "strawberry", "pepper", "pear", "peach", "papaya", "watermelon", "grape", "banana", "cucumber"] and predicted_quality in ["Mild", "Rotten"]:
+            if predicted_name.lower() in ["kaki", "tomato", "strawberry", "peeper", "pear", "peach", "papaya", "watermelon", "grape", "banana", "cucumber"] and predicted_quality in ["Mild", "Rotten"]:
                 st.write("Segmentation of Defective Region:")
                 try:
                     predictor, cfg = load_detectron_model(predicted_name)