Update app.py

app.py

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