add subfolder

onnx_export.py

@@ -71,9 +71,14 @@ def convert_onnx(model_id: str, task: str, folder: str) -> List:
 
     validate_model_outputs(onnx_config, model, output, onnx_outputs, atol)
     print(f"All good, model saved at: {output}")
-
-    operations = [CommitOperationAdd(path_in_repo=file_name, path_or_fileobj=os.path.join(folder, file_name)) for file_name in os.listdir(folder)]
-
+    
+    n_files = len([name for name in os.listdir(folder) if os.path.isfile(os.path.join(folder, name)) and not name.startswith(".")])
+    
+    if n_files == 1:
+        operations = [CommitOperationAdd(path_in_repo=file_name, path_or_fileobj=os.path.join(folder, file_name)) for file_name in os.listdir(folder)]
+    else:
+        operations = [CommitOperationAdd(path_in_repo=os.path.join("onnx", file_name), path_or_fileobj=os.path.join(folder, file_name)) for file_name in os.listdir(folder)]
+    
     return operations