retromarz commited on
Commit
385cf4a
·
verified ·
1 Parent(s): 595f3d5

switched to dataset for json storage

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -63,12 +63,12 @@ def save_to_json(image_name: str, caption: str, caption_type: str, caption_lengt
63
  json.dump(data, f, indent=4)
64
  logger.info(f"Saved result to {OUTPUT_JSON_PATH}")
65
 
66
- # Upload to Space repository
67
  api.upload_file(
68
  path_or_fileobj=OUTPUT_JSON_PATH,
69
  path_in_repo="outputs/captions.json",
70
- repo_id=REPO_ID,
71
- repo_type="space",
72
  token=HF_TOKEN
73
  )
74
  logger.info(f"Uploaded {OUTPUT_JSON_PATH} to Space repository {REPO_ID}")
 
63
  json.dump(data, f, indent=4)
64
  logger.info(f"Saved result to {OUTPUT_JSON_PATH}")
65
 
66
+ # Upload to Dataset repository
67
  api.upload_file(
68
  path_or_fileobj=OUTPUT_JSON_PATH,
69
  path_in_repo="outputs/captions.json",
70
+ repo_id=retromarz/plavu_dataset,
71
+ repo_type="dataset",
72
  token=HF_TOKEN
73
  )
74
  logger.info(f"Uploaded {OUTPUT_JSON_PATH} to Space repository {REPO_ID}")