obsxrver commited on
Commit
e905aec
·
verified ·
1 Parent(s): 08f7815

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -268,10 +268,11 @@ def hf_upload(file_path, prompt, repo):
268
  unique_name = str(uuid.uuid4())
269
  video_name=f"{unique_name}.mp4"
270
  caption_name=f"{unique_name}.txt"
 
271
 
272
  api.upload_file(
273
  path_or_fileobj=file_path,
274
- path_in_repo=video_name,
275
  repo_id=repo,
276
  repo_type="dataset"
277
  )
@@ -279,7 +280,7 @@ def hf_upload(file_path, prompt, repo):
279
  f.write(prompt)
280
  api.upload_file(
281
  path_or_fileobj=caption_name,
282
- path_in_repo=caption_name,
283
  repo_id=repo,
284
  repo_type="dataset"
285
  )
 
268
  unique_name = str(uuid.uuid4())
269
  video_name=f"{unique_name}.mp4"
270
  caption_name=f"{unique_name}.txt"
271
+ bucket =f"{unique_name[0]}/{unique_name[1]}/{unique_name[2]}"
272
 
273
  api.upload_file(
274
  path_or_fileobj=file_path,
275
+ path_in_repo=f"{bucket}/{video_name}",
276
  repo_id=repo,
277
  repo_type="dataset"
278
  )
 
280
  f.write(prompt)
281
  api.upload_file(
282
  path_or_fileobj=caption_name,
283
+ path_in_repo=f"{bucket}/{caption_name}",
284
  repo_id=repo,
285
  repo_type="dataset"
286
  )