test_pred

app.py

@@ -17,12 +17,11 @@ option = st.selectbox('Which pre-trained model would you like for your sentiment
 st.write('You selected:', option)
 
 if option == 'MILESTONE 3':
-    st.write('test1')
-    # model_name_0 = "Rathgeberj/milestone3_0"
-    # # model_0 = AutoModelForSequenceClassification.from_pretrained(model_name_0)
-    # model_0 = BertForMaskedLM.from_pretrained(model_name_0)
-    # tokenizer_0 = AutoTokenizer.from_pretrained(model_name_0)
-    # classifier_0 = pipeline(task="sentiment-analysis", model=model_0, tokenizer=tokenizer_0)
+    model_name_0 = "Rathgeberj/milestone3_0"
+    # model_0 = AutoModelForSequenceClassification.from_pretrained(model_name_0)
+    model_0 = BertForMaskedLM.from_pretrained(model_name_0)
+    tokenizer_0 = AutoTokenizer.from_pretrained(model_name_0)
+    classifier_0 = pipeline(task="sentiment-analysis", model=model_0, tokenizer=tokenizer_0)
 
     # model_name_1 = "Rathgeberj/milestone3_1"
     # # model_1 = AutoModelForSequenceClassification.from_pretrained(model_name_1)
@@ -58,8 +57,18 @@ if option == 'MILESTONE 3':
     # tokenizers = [tokenizer_0, tokenizer_1, tokenizer_2, tokenizer_3, tokenizer_4, tokenizer_5]
     # classifiers = [classifier_0, classifier_1, classifier_2, classifier_3, classifier_4, classifier_5]
 
-    # X_train = [textIn]
-    # batch = tokenizer(X_train, padding=True, truncation=True, max_length=512, return_tensors="pt")
+    X_train = [textIn]
+    batch = tokenizer(X_train, padding=True, truncation=True, max_length=512, return_tensors="pt")
+
+    with torch.no_grad():
+        outputs = model(**batch, labels=torch.tensor([1, 0]))
+        predictions = F.softmax(outputs.logits, dim=1)
+        labels = torch.argmax(predictions, dim=1)
+        labels = [model.config.id2label[label_id] for label_id in labels.tolist()]
+
+    st.write(outputs['label'])
+
+
     col = ['Tweet', 'Highest_Toxicity_Class_Overall', 'Score_Overall', 'Highest_Toxicity_Class_Except_Toxic', 'Score_Except_Toxic']
     df = pd.DataFrame(columns=col)
     pre_populated_tweets = ['Yo bitch Ja Rule is more succesful then youll ever be whats up with you and hating you sad mofuckas...i should bitch slap ur pethedic white faces and get you to kiss my ass you guys sicken me. Ja rule is about pride in da music man. dont diss that shit on him. and nothin is wrong bein like tupac he was a brother too...fuckin white boys get things right next time.',