21 lines
429 B
Python
21 lines
429 B
Python
|
import pandas as pd
|
||
|
|
import numpy as np
|
||
|
|
import os
|
||
|
|
import io
|
||
|
|
import s3fs
|
||
|
|
import re
|
||
|
|
import pickle
|
||
|
|
import warnings
|
||
|
|
|
||
|
|
|
||
|
|
def load_model(type_of_activity, model):
|
||
|
|
BUCKET = f"projet-bdc2324-team1/Output_model/{type_of_activity}/{model}/"
|
||
|
|
filename = model + '.pkl'
|
||
|
|
file_path = BUCKET + filename
|
||
|
|
with fs.open(file_path, mode="rb") as f:
|
||
|
|
model_bytes = f.read()
|
||
|
|
|
||
|
|
model = pickle.loads(model_bytes)
|
||
|
|
return model
|
||
|
|
|