Update app.py

app.py

@@ -109,11 +109,12 @@ def predict(user_input_text):
 
     response += "<br>TOP 2 MOST INFLUENTIAL WORDS FOR EACH LABEL:<br>"
     for i, explanations in enumerate(explanation_lists):
-        label = labels[i]
-        response += f"{label}:<br>"
-        for explanation in explanations:
-            if explanation.explainer == 'Partition SHAP':
-                sorted_scores = sorted(enumerate(explanation.scores), key=lambda x: abs(x[1]), reverse=True)[:2]
+    label = labels[i]
+    response += f"{label}:<br>"
+    for explanation in explanations:
+        if explanation.explainer == 'Partition SHAP':
+            sorted_scores = sorted(enumerate(explanation.scores), key=lambda x: abs(x[1]), reverse=True)[:2]
+            if sorted_scores:  # Check if the list is not empty
                 tokens = replace_encoding(explanation.tokens)
                 tokens = [tokens[idx] for idx, _ in sorted_scores]
                 formatted_output = ' '.join(tokens)