[fix] background without leaves causing error due to parsing

app.py

@@ -27,9 +27,16 @@ def predict(image):
     label = model.config.id2label[label_index]
 
     # Split the label into leaf category and disease name
+
     label_parts = label.split("___")
     leaf_category = label_parts[0]
-    disease_name = label_parts[1]
+
+    disease_name: str = ""
+    if len(label_parts) == 1:
+        # if label parts only has one part, assume "background with leaves" as the category and "N/A as the disease name"
+        disease_name = "N/A"
+    else:
+        disease_name = label_parts[1]
 
     # Calculate the percentage breakdown of predicted diseases
     percentage_breakdown = {disease: round(float(probability[0, index]) * 100, 2) for index, disease in enumerate(model.config.label2id)}