Update upload-onnx.py
Browse files- upload-onnx.py +6 -5
upload-onnx.py
CHANGED
@@ -16,12 +16,12 @@ def main():
|
|
16 |
onnx_json_file = f"{model_name}.onnx.json"
|
17 |
|
18 |
# Verifica si el archivo .onnx existe
|
19 |
-
if os.path.isfile(onnx_file):
|
20 |
-
print(f"{model_name}.onnx encontrado.")
|
21 |
-
else:
|
22 |
print(f"{model_name}.onnx no se encuentra.")
|
23 |
return
|
24 |
|
|
|
|
|
25 |
# Renombra config.json a ${MODELNAME}.onnx.json si existe
|
26 |
if os.path.isfile(config_file):
|
27 |
shutil.move(config_file, onnx_json_file)
|
@@ -49,8 +49,9 @@ def main():
|
|
49 |
# Crea el repositorio en Hugging Face si no existe
|
50 |
repo_id = f"{username}/{model_name}"
|
51 |
try:
|
52 |
-
|
53 |
-
|
|
|
54 |
except Exception as e:
|
55 |
print(f"El repositorio {repo_id} ya existe o se produjo un error: {e}")
|
56 |
|
|
|
16 |
onnx_json_file = f"{model_name}.onnx.json"
|
17 |
|
18 |
# Verifica si el archivo .onnx existe
|
19 |
+
if not os.path.isfile(onnx_file):
|
|
|
|
|
20 |
print(f"{model_name}.onnx no se encuentra.")
|
21 |
return
|
22 |
|
23 |
+
print(f"{model_name}.onnx encontrado.")
|
24 |
+
|
25 |
# Renombra config.json a ${MODELNAME}.onnx.json si existe
|
26 |
if os.path.isfile(config_file):
|
27 |
shutil.move(config_file, onnx_json_file)
|
|
|
49 |
# Crea el repositorio en Hugging Face si no existe
|
50 |
repo_id = f"{username}/{model_name}"
|
51 |
try:
|
52 |
+
# Crear el repositorio como privado
|
53 |
+
api.create_repo(repo_id=repo_id, token=token, private=True)
|
54 |
+
print(f"Repositorio {repo_id} creado en Hugging Face como privado.")
|
55 |
except Exception as e:
|
56 |
print(f"El repositorio {repo_id} ya existe o se produjo un error: {e}")
|
57 |
|