File tree 2 files changed +5
-1
lines changed
2 files changed +5
-1
lines changed Original file line number Diff line number Diff line change 78
78
16 : exceptions .UNAUTHENTICATED ,
79
79
}
80
80
81
+
81
82
def _get_initialized_app (app ):
82
83
if app is None :
83
84
return firebase_admin .get_app ()
Original file line number Diff line number Diff line change 22
22
import datetime
23
23
import re
24
24
import time
25
+ import os
25
26
import requests
26
27
import six
27
28
@@ -440,8 +441,10 @@ def _parse_gcs_tflite_uri(uri):
440
441
def upload (bucket_name , model_file_name , app ):
441
442
"""Upload a model file to the specified Storage bucket."""
442
443
_CloudStorageClient ._assert_gcs_enabled ()
444
+
445
+ file_name = os .path .basename (model_file_name )
443
446
bucket = storage .bucket (bucket_name , app = app )
444
- blob_name = _CloudStorageClient .BLOB_NAME .format (model_file_name )
447
+ blob_name = _CloudStorageClient .BLOB_NAME .format (file_name )
445
448
blob = bucket .blob (blob_name )
446
449
blob .upload_from_filename (model_file_name )
447
450
return _CloudStorageClient .GCS_URI .format (bucket .name , blob_name )
You can’t perform that action at this time.
0 commit comments