source/plant.py

@@ -24,12 +24,8 @@ 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":