def get_model(bucket, team_id, user_id):
key = f"models/{team_id}/{user_id}.json"
print(f'downloading model {key}')
model_json = BytesIO()
bucket.download_fileobj(key, model_json)
model = str(model_json.getbuffer(), 'utf-8')
return NewlineText.from_chain(model)
def get_compressed_model(bucket, team_id, user_id):
key = f"models/{team_id}/{user_id}.gz"
print(f'downloading model {key}')
model_gz = BytesIO()
bucket.download_fileobj(key, model_gz)
model = str(zlib.decompress(model_gz.getbuffer()), 'utf-8')
return NewlineText.from_chain(model)