Update app/main.py

app/main.py

@@ -100,7 +100,9 @@ async def moderate_text(request: TextRequest):
     text = request.text
 
     profanity_result = profanity_filter.check_profanity(text)
-    tox_result = toxicity_pipe(text)[0]
+    # tox_result = toxicity_pipe(text)[0]
+    raw_result = toxicity_pipe(text)
+    tox_result = raw_result[0][0] if isinstance(raw_result[0], list) else raw_result[0]
 
     is_toxic_by_model = (
         tox_result["label"].lower() in ["toxic", "label_1"] and tox_result["score"] > 0.5