Update save_model_file_path_to_env.py

Signed-off-by: Thanawan Atchariyachanvanit <[email protected]>
thanawan-atc · Jul 12, 2023 · 5de533a · 5de533a
1 parent 4f5e91f
commit 5de533a
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/utils/model_uploader/save_model_file_path_to_env.py b/utils/model_uploader/save_model_file_path_to_env.py
@@ -10,9 +10,9 @@
 def get_model_file_path(model_id, model_version, model_format):
     model_name = str(model_id.split("/")[-1])
     model_format = model_format.lower()
-    model_dirname = f"{model_name}/{model_version}/{model_format}"
+    model_dirname = f"sentence-transformers/upload/{model_name}/{model_version}/{model_format}"
     model_filename = (
-        f"sentence-transformers/upload/sentence-transformers_{model_name}-{model_version}-{model_format}.zip"
+        f"sentence-transformers_{model_name}-{model_version}-{model_format}.zip"
     )
     model_file_path = model_dirname + "/" + model_filename
     return model_file_path