Update app.py

app.py

@@ -70,16 +70,10 @@ def replace_encoding(tokens):
             for token in tokens[1:-1]]
 
 def predict(text):
-
-    explanations_formality = exp(text, target=0)
-    explanations_sentiment = exp(text, target=1)
-    explanations_politeness = exp(text, target=2)
-    explanations_toxicity = exp(text, target=3)
-    
-    #explanations_formality = bench.explain(text, target=0)
-    #explanations_sentiment = bench.explain(text, target=1)
-    #explanations_politeness = bench.explain(text, target=2)
-    #explanations_toxicity = bench.explain(text, target=3)
+    explanations_formality = [bench.explain(text, target=0)]
+    explanations_sentiment = [bench.explain(text, target=1)]
+    explanations_politeness = [bench.explain(text, target=2)]
+    explanations_toxicity = [bench.explain(text, target=3)]
     
     greeting_pattern = r"^(Halló|Hæ|Sæl|Góðan dag|Kær kveðja|Daginn|Kvöldið|Ágætis|Elsku)"
     
@@ -95,7 +89,6 @@ def predict(text):
 
     response = f"INPUT:<br>{modified_input}<br><br>MY PREDICTION:<br>{prediction_output}<br>{influential_keywords}<br>{greeting_feedback}"
 
-    # Include influential words in the response
     explanation_lists = [explanations_toxicity, explanations_formality, explanations_sentiment, explanations_politeness]
     labels = ['Toxicity', 'Formality', 'Sentiment', 'Politeness']
 
@@ -109,20 +102,9 @@ def predict(text):
                 formatted_output = ' '.join([f"{token} ({score})" for token, score in token_score_pairs])
                 response += f"{label}: {formatted_output}<br>"
 
-    #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]
-    #            tokens = replace_encoding(explanation.tokens)
-    #            tokens = [tokens[idx] for idx, _ in sorted_scores]
-    #            formatted_output = ' '.join(tokens)
-    #            response += f"{formatted_output}<br>"
-
     return response
 
+
 description_html = """
     <center>
         <img src='http://www.ru.is/media/HR_logo_vinstri_transparent.png' width='250' height='auto'>