source/plant.py

@@ -24,18 +24,14 @@ class Plant:
             self.name = "Apple"
             self.plant_type = "fruit"
 
-        elif predicted_class == "Strawberry":
-            self.name = "Strawberry"
-            self.plant_type = "fruit"
-
-        elif predicted_class == "Cucumber":
-            self.name = "Cucumber"
+        elif predicted_class == "Radish":
+            self.name = "Radish"
             self.plant_type = "vegetable"
 
         elif predicted_class == "Cauliflower":
             self.name = "Cauliflower"
             self.plant_type = "vegetable"
-            
+
         elif predicted_class == "Wheat":
             self.name = "Wheat"
             self.plant_type = "cereal"