Update app.py

app.py

@@ -22,7 +22,7 @@ def get_cats(text):
   model = AutoModelForSequenceClassification.from_pretrained("aim9061/fine-tuned-toxic-tweet-dilbert")
   outputs = model(**token)
   
-  prob = torch.sigmoid(output.logits).detach().numpy()[0]
+  prob = torch.sigmoid(outputs.logits).detach().numpy()[0]
   
   data = create_dict(prob, text)
   st.table(data)