Update app.py

app.py

@@ -71,7 +71,7 @@ from PyPDF2 import PdfReader
 from huggingface_hub import HfApi
 import io
 from datasets import load_dataset
-import dill
+import joblib
 
 # from huggingface_hub import Repository
 # from git import repo
@@ -169,11 +169,13 @@ def main():
       class_list.append(i['word'])
       entity_list.append(i['entity_group'])
   
-  filename = 'Checkpoint-classification.sav'
+  #filename = 'Checkpoint-classification.sav'
+  filename = 'model.bin'
   count_vect = CountVectorizer(ngram_range=[1,3])
   tfidf_transformer=TfidfTransformer()
   #loaded_model = pickle.load(open(filename, 'rb'))
-  loaded_model = pickle.load(open(filename, 'rb'))
+  #loaded_model = pickle.load(open(filename, 'rb'))
+  loaded_model = joblib.load(filename)
   #loaded_vectorizer = dill.load(open('vectorizefile_classification.pickle', 'rb'))
   #loaded_vectorizer = pickle.load(open('vectorizefile_classification.pickle', 'rb'))