testfunc

app.py

@@ -5,6 +5,9 @@ from textblob import TextBlob
 from transformers import AutoTokenizer, AutoModelForSequenceClassification
 import torch
 import torch.nn.functional as F
+from transformers import BertForMaskedLM
+
+model = BertForMaskedLM.from_pretrained("remi/bertabs-finetuned-extractive-abstractive-summarization")
 
 textIn = st.text_input("Input Text Here:", "I really like the color of your car!")
 
@@ -16,8 +19,9 @@ if option == 'MILESTONE 3':
 
     st.write('test1')
 
-    model_name_0 = "milestone3_0"
-    model_0 = AutoModelForSequenceClassification.from_pretrained(model_name_0)
+    model_name_0 = "Rathgeberj/milestone3_1"
+    # model_0 = AutoModelForSequenceClassification.from_pretrained(model_name_0)
+    model_0 = BertForMaskedLM.from_pretrained("Rathgeberj/milestone3_0")
     tokenizer_0 = AutoTokenizer.from_pretrained(model_name_0)
     classifier_0 = pipeline(task="sentiment-analysis", model=model_0, tokenizer=tokenizer_0)