Spaces:
Running
Running
| import pandas as pd | |
| import joblib | |
| import config | |
| def load_model(): | |
| return joblib.load(config.MODEL_PATH) | |
| def load_ppp_rpp(): | |
| ppp = pd.read_csv(config.PPP_PATH, index_col=0) | |
| rpp = pd.read_csv(config.RPP_PATH, index_col=0) | |
| return ppp, rpp | |
| def load_lambdas(): | |
| df = pd.read_csv(config.LAMBDAS_PATH) | |
| return dict(zip(df['key'], df['value'])) | |
| def load_locations(): | |
| countries = pd.read_csv(config.COUNTRIES_PATH) | |
| states = pd.read_csv(config.STATES_PATH) | |
| countries_dict = dict(zip(countries['alpha-2'], countries['name'])) | |
| states_dict = dict(zip(states['Abbreviation'], states['State'])) | |
| return countries_dict, states_dict |