fixed some issues w/ the uncertainty

api/predict.py

@@ -175,7 +175,7 @@ def ensemblePredict(text):
         
         finalPred = 1 if isFake else 0
         agreementCount = sum(1 for p in allPreds if p == finalPred)
-        agreement = float(agreementCount) / len(allPreds)
+        agreement = agreementCount / len(allPreds)
         
         del weightedProbs, allPreds
         gc.collect()
@@ -264,23 +264,23 @@ def predict_review(text):
     
     confidence = max(fakeProb, genuineProb)
     
+    prediction = "fake" if isFake else "genuine"
+    isFakeOutput = isFake
+    
     if agreement < uncertaintyThreshold:
         prediction = "uncertain"
         isFakeOutput = None
-    else:
-        prediction = "fake" if isFake else "genuine"
-        isFakeOutput = isFake
     
     lengthCat = getLengthCategory(cleaned)
     tokenCount = len(cleaned.split())
     
     return {
         "prediction": prediction,
-        "confidence": float(confidence),
+        "confidence": confidence,
         "is_fake": isFakeOutput,
-        "model_agreement": float(agreement),
-        "fake_probability": float(fakeProb),
-        "genuine_probability": float(genuineProb),
+        "model_agreement": agreement,
+        "fake_probability": fakeProb,
+        "genuine_probability": genuineProb,
         "length_category": lengthCat,
         "token_count": tokenCount
     }