Spaces:
Runtime error
Runtime error
import pandas as pd | |
import numpy as np | |
import pickle | |
import warnings | |
warnings.filterwarnings("ignore") | |
def vectorizer(): | |
category_mapping = {'Debt collection': 0, | |
'Mortgage': 1, | |
'Consumer Loan': 2, | |
'Bank account or service': 3, | |
'Credit reporting': 4, | |
'Payday loan': 5, | |
'Other financial service': 6, | |
'Student loan': 7, | |
'Money transfers': 8, | |
'Prepaid card': 9, | |
'Credit card': 10 | |
} | |
print("Created Mapping") | |
model = pickle.load(open("model.pkl",'rb')) | |
vectorizer = pickle.load(open("vectorizer.pkl",'rb')) | |
print("Loaded the model") | |
return model, vectorizer, category_mapping |