feat: Add produkt feedback detection

app.py

@@ -96,24 +96,46 @@ def classification(task: str, doc: str) -> str:
     # Else if the task is spam detection, classify the text into spam or not spam
     elif task == "Spam detection":
         if language == "sv":
-            hypothesis_template = "Det här e-postmeddelandet ser {}"
+            hypothesis_template = "Det här e-postmeddelandet ser {}."
             candidate_labels = {
                 "ut som ett skräppostmeddelande": "Spam",
                 "inte ut som ett skräppostmeddelande": "Inte spam",
             }
         elif language == "no":
-            hypothesis_template = "Denne e-posten ser {}"
+            hypothesis_template = "Denne e-posten ser {}."
             candidate_labels = {
                 "ut som en spam-e-post": "Spam",
                 "ikke ut som en spam-e-post": "Ikke spam",
             }
         else:
-            hypothesis_template = "Denne e-mail ligner {}"
+            hypothesis_template = "Denne e-mail ligner {}."
             candidate_labels = {
                 "en spam e-mail": "Spam",
                 "ikke en spam e-mail": "Ikke spam",
             }
 
+    # Else if the task is product feedback detection, classify the text into product
+    # feedback or not product feedback
+    elif task == "Product feedback detection":
+        if language == "sv":
+            hypothesis_template = "Den här kommentaren är {}."
+            candidate_labels = {
+                "en recension av en produkt": "Produktfeedback",
+                "inte en recension av en produkt": "Inte produktfeedback",
+            }
+        elif language == "no":
+            hypothesis_template = "Denne kommentaren er {}."
+            candidate_labels = {
+                "en anmeldelse av et produkt": "Produkttilbakemelding",
+                "ikke en anmeldelse av et produkt": "Ikke produkttilbakemelding",
+            }
+        else:
+            hypothesis_template = "Denne kommentar er {}."
+            candidate_labels = {
+                "en anmeldelse af et produkt": "Produktfeedback",
+                "ikke en anmeldelse af et produkt": "Ikke produktfeedback",
+            }
+
     # Else the task is not supported, so raise an error
     else:
         raise ValueError(f"Task {task} not supported.")
@@ -141,7 +163,12 @@ def classification(task: str, doc: str) -> str:
 # Create a dropdown menu for the task
 dropdown = gr.inputs.Dropdown(
     label="Task",
-    choices=["Sentiment classification", "News topic classification", "Spam detection"],
+    choices=[
+        "Sentiment classification",
+        "News topic classification",
+        "Spam detection",
+        "Product feedback detection",
+    ],
     default="Sentiment classification",
 )