Update app.py

app.py

@@ -5,7 +5,7 @@ import re
 text_pipe = pipeline("text-classification", model="karalif/myTestModel", return_all_scores=True)
 
 def predict(text):
-    greeting_pattern = r"^(Halló|Hæ|Sæl|Góða|Kær|Daginn|Kvöldið|Ágæt|Elsku)"
+    greeting_pattern = r"^(Halló|Hæ|Sæl|Góðan dag|Kær kveðja|Daginn|Kvöldið|Ágætis|Elsku)"
     
     greeting_feedback = ""
     
@@ -15,16 +15,14 @@ def predict(text):
     for result in all_scores:
         label = result['label']
         score = result['score']
-        if label in ["Politeness"]:
+        if label == "Politeness":
             response += f"<span style='background-color:#b8e994; color:black;'>{label}</span>: {score:.3f}<br>"  # Light Green
-        if label in ["Sentiment"]:
+        elif label == "Sentiment":
             response += f"<span style='background-color:#fff3cd; color:black;'>{label}</span>: {score:.3f}<br>"  # Light Yellow
-        if label in ["Formality"]:
+        elif label == "Formality":
             response += f"<span style='background-color:#bee5eb; color:black;'>{label}</span>: {score:.3f}<br>"  # Light Blue
-        if label in ["Toxicity"]:
+        elif label == "Toxicity":
             response += f"<span style='background-color:#f8d7da; color:black;'>{label}</span>: {score:.3f}<br>"  # Light Red
-        else:
-            response += f"{label}: {score:.3f}<br>"
     
     if not re.match(greeting_pattern, text, re.IGNORECASE):
         greeting_feedback = "Heilsaðu dóninn þinn<br>"
@@ -33,6 +31,7 @@ def predict(text):
     
     return response
 
+
 description_html = """
     <center>
         <img src='http://www.ru.is/media/HR_logo_vinstri_transparent.png' width='250' height='auto'>