Update textPreprocess.py

textPreprocess.py

@@ -99,9 +99,13 @@ def predict_text(text: str, max_length: int = None):
         pred = predictions[0]
         # pred.label is "Human" or "AI"
         # Map to "Human" or "Ai" to match previous API
-        label = pred.label
-        if label == "AI":
-            label = "Ai"
+        # Determine label based on higher confidence
+        if pred.probability_ai > pred.probability_human:
+            label = "AI"
+            confidence = pred.probability_ai
+        else:
+            label = "Human"
+            confidence = pred.probability_human
         
         # Confidence logic:
         # If label is Human, use probability_human
@@ -141,9 +145,15 @@ def predict_batch(texts, batch_size=16):
         results = []
         for text, pred in zip(texts, predictions):
             label = pred.label
-            if label == "AI":
-                label = "Ai"
-            confidence = pred.probability_human if label == "Human" else pred.probability_ai
+            
+            # Determine label based on higher confidence
+            if pred.probability_ai > pred.probability_human:
+                label = "AI"
+                confidence = pred.probability_ai
+            else:
+                label = "Human"
+                confidence = pred.probability_human
+
             
             results.append({
                 "text": text,