Update app.py

app.py

@@ -87,9 +87,10 @@ async def predict_text_endpoint(payload: TextPredictionRequest = Body(...)):
         
         return TextPredictionResponse(
             predicted_class=result["predicted_class"], 
-            confidence=result["confidence"]
+            confidence_ai=result["confidence_ai"],
+            confidence_human=result["confidence_human"]
         )
     except Exception as e:
         # Return a fallback response in case of error
         print(f"Error in text prediction: {e}")
-        return TextPredictionResponse(predicted_class="Human", confidence=0.5)
+        return TextPredictionResponse(predicted_class="Human", confidence_ai=-100.0, confidence_human=-100.0)