Update pages/core_risk.py

pages/core_risk.py

@@ -1,19 +1,12 @@
 import numpy as np
-import torch
 import transformers
 import streamlit as st
 from streamlit import session_state
 import json
-import torch.nn.functional as F
-import boto3
 import pandas as pd
-bucket = 'data-ai-dev2'
 from transformers import BertTokenizer, BertModel
 from torch import cuda
-device = 'cuda' if cuda.is_available() else 'cpu'
 import numpy
-from numpy.random import seed
-seed(1)
 import emoji
 import string
 import nltk
@@ -34,13 +27,7 @@ from transformers import pipeline
 stopwords = nltk.corpus.stopwords.words('english')
 
 
-model = 'C:/Users/Meet/Downloads/core_risk/models/'
-tokenizer = 'C:/Users/Meet/Downloads/core_risk/tokenizer/'
-
-
-from transformers import pipeline
-
-classifier = pipeline("text-classification", model= model,  tokenizer = tokenizer, truncation=True, max_length=512)
+classifier = pipeline("text-classification", model="dsmsb/16class_12k_newtest1618_xlm_roberta_base_27nov_v2_8epoch")
 def pre_processing_str_esg(df_col):
     df_col = df_col.lower()
     #defining the function to remove punctuation