diff --git a/firebase_admin/_utils.py b/firebase_admin/_utils.py index 495632ad2..e1eb83d90 100644 --- a/firebase_admin/_utils.py +++ b/firebase_admin/_utils.py @@ -78,6 +78,7 @@ 16: exceptions.UNAUTHENTICATED, } + def _get_initialized_app(app): if app is None: return firebase_admin.get_app() diff --git a/firebase_admin/ml.py b/firebase_admin/ml.py index d34bdd581..fa604933c 100644 --- a/firebase_admin/ml.py +++ b/firebase_admin/ml.py @@ -22,6 +22,7 @@ import datetime import re import time +import os import requests import six @@ -440,8 +441,10 @@ def _parse_gcs_tflite_uri(uri): def upload(bucket_name, model_file_name, app): """Upload a model file to the specified Storage bucket.""" _CloudStorageClient._assert_gcs_enabled() + + file_name = os.path.basename(model_file_name) bucket = storage.bucket(bucket_name, app=app) - blob_name = _CloudStorageClient.BLOB_NAME.format(model_file_name) + blob_name = _CloudStorageClient.BLOB_NAME.format(file_name) blob = bucket.blob(blob_name) blob.upload_from_filename(model_file_name) return _CloudStorageClient.GCS_URI.format(bucket.name, blob_name)