Test assertion provided

app.py

@@ -25,9 +25,11 @@ def classify_text(text):
     for pred in [roberta_base_pred, chatgpt_lli_hc3_pred, chatgpt_roberta_pred]:
         if pred == 1:
             votes["AI"] += 1
-        else:
+        elif pred == 0:
             votes["Human"] += 1
-
+        else:
+            raise AssertionError("A problem exists with the code.")
+            
     # Determine final decision based on majority
     if votes["AI"] > votes["Human"]:
         return "AI"