Spaces:
Sleeping
Sleeping
File size: 685 Bytes
f4642b6 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 | 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 |