Update app.py

app.py

@@ -2,6 +2,10 @@ import gradio as gr
 import transformers
 import torch
 from transformers import BertModel, BertTokenizer, AdamW, get_linear_schedule_with_warmup
+from torch import nn, optim
+from torch.utils.data import Dataset, DataLoader
+import pickle
+
 
 class_names = ['left', 'neutral', 'right']
 PRE_TRAINED_MODEL_NAME = 'bert-base-uncased'
@@ -27,6 +31,8 @@ class SentimentClassifier(nn.Module):
 
 model = SentimentClassifier(len(class_names))
 
+
+model2 = pickle.load(open("model_BERT_2", "rb"))
 def result_final(new_article):
 
   encoded_review = tokenizer.encode_plus(
@@ -50,4 +56,5 @@ def result_final(new_article):
 
 iface = gr.Interface(fn = result_final, inputs = "text", outputs = ["text"], title = "News Bias Classifer")
 iface.launch()
+