tomo2chin2 commited on
Commit
aeb151c
·
verified ·
1 Parent(s): f68c84d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +20 -3
app.py CHANGED
@@ -1,3 +1,4 @@
 
1
  import os
2
  import json
3
  import re
@@ -78,18 +79,34 @@ def download_and_upload(model_id, repo_type):
78
  "sequence_number": sequence_number
79
  }
80
 
 
 
 
 
 
81
  # ファイルのアップロード
82
  try:
 
83
  api.upload_file(
84
  path_or_fileobj=filename,
85
  path_in_repo=filename,
86
  repo_id=target_repo,
87
  token=HF_TOKEN,
88
  repo_type="dataset",
89
- commit_message=f"Add {filename} from model ID {model_id}",
90
- commit_description=json.dumps(metadata)
 
 
 
 
 
 
 
 
 
91
  )
92
- return f"ファイル '{filename}' を '{target_repo}' にアップロードしました。"
 
93
  except Exception as e:
94
  return f"アップロード中にエラーが発生しました: {e}"
95
 
 
1
+ # メタデータを含めるためのライブラリをインポート
2
  import os
3
  import json
4
  import re
 
79
  "sequence_number": sequence_number
80
  }
81
 
82
+ # メタデータをJSONファイルとして保存
83
+ metadata_filename = f"{filename}.metadata.json"
84
+ with open(metadata_filename, "w") as meta_file:
85
+ json.dump(metadata, meta_file)
86
+
87
  # ファイルのアップロード
88
  try:
89
+ # 元のファイルをアップロード
90
  api.upload_file(
91
  path_or_fileobj=filename,
92
  path_in_repo=filename,
93
  repo_id=target_repo,
94
  token=HF_TOKEN,
95
  repo_type="dataset",
96
+ commit_message=f"Add {filename} from model ID {model_id}"
97
+ )
98
+
99
+ # メタデータファイルをアップロード
100
+ api.upload_file(
101
+ path_or_fileobj=metadata_filename,
102
+ path_in_repo=metadata_filename,
103
+ repo_id=target_repo,
104
+ token=HF_TOKEN,
105
+ repo_type="dataset",
106
+ commit_message=f"Add metadata for {filename}"
107
  )
108
+
109
+ return f"ファイル '{filename}' とメタデータを '{target_repo}' にアップロードしました。"
110
  except Exception as e:
111
  return f"アップロード中にエラーが発生しました: {e}"
112