Getting logits

app.py

@@ -32,8 +32,8 @@ def moderate_image(img):
 
 def classify_toxic(text):
     batch = tokenizer.encode(text, return_tensors="pt")
-    output = model(batch).logits
-    print(output)
+    output = model(batch).logits.detach().softmax(dim=-1)
+    print(f"Output: {output}")
     return output