Updated the structure

app.py

@@ -81,9 +81,18 @@ async def predict(
             if plant == "tomato" and raw_label.startswith("Tomato___"):
                 label = raw_label.replace("Tomato___", "")
                 if label not in TOMATO_CLASSES:
-                    label = raw_label  # fallback
+                     # Remove everything before "___" if present
+                    if "___" in raw_label:
+                        label = raw_label.split("___", 1)[1]
+                    else:
+                        label = raw_label
+
             else:
-                label = model_score
+                # Remove everything before "___" if present
+                if "___" in raw_label:
+                    label = raw_label.split("___", 1)[1]
+                else:
+                    label = raw_label
 
             mapped_results.append({
                 "class": label,