Update Sentiment_analysis_with_bert.py

Sentiment_analysis_with_bert.py

@@ -22,7 +22,15 @@ from torch import nn, optim
 from torch.utils.data import Dataset, DataLoader
 import torch.nn.functional as F
 
+from transformers import BertModel
+
+bert_model = BertModel.from_pretrained('bert-base-uncased')
+
+bert_model.save_pretrained('C:/Users/Marie-Ange/Downloads')
 
+from transformers import pipeline
+pipe = pipeline(model="Group209/Sentiment_Analysis")
+pipe("Sentiment_analysis_with_bert.py")
 
 sns.set(style='whitegrid', palette='muted', font_scale=1.2)
 
@@ -520,8 +528,4 @@ def generate_improved_text(review_text, predicted_sentiment):
 
     return improved_text
 
-from transformers import BertModel
-
-bert_model = BertModel.from_pretrained('bert-base-uncased')
 
-bert_model.save_pretrained('C:/Users/Marie-Ange/Downloads')