Update upload-onnx.py

upload-onnx.py

@@ -16,12 +16,12 @@ def main():
     onnx_json_file = f"{model_name}.onnx.json"
 
     # Verifica si el archivo .onnx existe
-    if os.path.isfile(onnx_file):
-        print(f"{model_name}.onnx encontrado.")
-    else:
+    if not os.path.isfile(onnx_file):
         print(f"{model_name}.onnx no se encuentra.")
         return
 
+    print(f"{model_name}.onnx encontrado.")
+
     # Renombra config.json a ${MODELNAME}.onnx.json si existe
     if os.path.isfile(config_file):
         shutil.move(config_file, onnx_json_file)
@@ -49,8 +49,9 @@ def main():
     # Crea el repositorio en Hugging Face si no existe
     repo_id = f"{username}/{model_name}"
     try:
-        api.create_repo(repo_id=repo_id, token=token)
-        print(f"Repositorio {repo_id} creado en Hugging Face.")
+        # Crear el repositorio como privado
+        api.create_repo(repo_id=repo_id, token=token, private=True)
+        print(f"Repositorio {repo_id} creado en Hugging Face como privado.")
     except Exception as e:
         print(f"El repositorio {repo_id} ya existe o se produjo un error: {e}")