Update app.py

app.py

@@ -70,6 +70,7 @@ from PyPDF2 import PdfReader
 from huggingface_hub import HfApi
 import io
 from datasets import load_dataset
+import dill
 
 # from huggingface_hub import Repository
 # from git import repo
@@ -171,7 +172,8 @@ def main():
   count_vect = CountVectorizer(ngram_range=[1,3])
   tfidf_transformer=TfidfTransformer()
   loaded_model = pickle.load(open(filename, 'rb'))
-  loaded_vectorizer = pickle.load(open('vectorizefile_classification.pickle', 'rb'))
+  loaded_vectorizer = dill.load(open('vectorizefile_classification.pickle', 'rb'))
+  #loaded_vectorizer = pickle.load(open('vectorizefile_classification.pickle', 'rb'))
 
   pipeline_test_output = loaded_vectorizer.transform(class_list)
   predicted = loaded_model.predict(pipeline_test_output)