Testing NLP side

app.py

@@ -33,9 +33,8 @@ def moderate_image(img):
 def classify_toxic(text):
     batch = tokenizer.encode(text, return_tensors="pt")
     output = model(batch)
-    output = output[0].tolist()
-    max_idx = max(output)
-    return "Toxic" if output.index(max_idx) == 1 else "Neutral"
+    print(output)
+    return output
 
 
 # -----------------------
@@ -110,7 +109,7 @@ body, .gradio-container {
 .footer {
     flex-shrink: 0;
     text-align: center;
-    font-size: 0.85rem;
+    font-size: 0.80rem;
     opacity: 0.6;
     padding: 14px 0;
     border-top: 1px solid rgba(var(--block-border-color-rgb), 0.12);
@@ -164,4 +163,4 @@ with gr.Blocks(
     gr.Markdown("<div class='footer'>Demo by 7th • Powered by Transformers</div>")
 
 if __name__ == "__main__":
-    demo.launch(share=True)
+    demo.launch(share=True, analytics_enabled=False)