Update app.py

app.py

@@ -3,7 +3,7 @@ import pandas as pd
 import numpy as np
 from sklearn.feature_extraction.text import CountVectorizer
 from sklearn.ensemble import RandomForestClassifier
-import pickle
+import joblib
 
 import warnings
 warnings.filterwarnings('ignore')
@@ -71,7 +71,7 @@ if uploaded_file is not None:
     df_ext = pd.concat([df, text_vectors_df],axis=1)
     df = df_ext.drop(['GMN','Product_Description','EA_GTIN','CV_GTIN'],axis=1)
 
-    loaded_model = pickle.load(open('rfc_model_grid.pkl', 'rb'))
+    loaded_model = joblib.load(open('rfc_model_grid.sav', 'rb'))
     result = loaded_model.predict(df)
     data['Product_Type_Predicted']=result