Update app.py

app.py

@@ -1,12 +1,12 @@
 import streamlit as st
 import torch
 from transformers import RobertaTokenizer
-from bert import RobertaClass
+from model import RobertaClass
 from text_preprocessing import preprocess_text
 
 # Load the fine-tuned BERT model
 model = RobertaClass()
-model.load_state_dict(torch.load('model_bert_2.bin',
+model.load_state_dict(torch.load('model.bin',
                                  map_location=torch.device('cpu')))
 
 # Load the tokenizer