fdaudens HF Staff commited on
Commit
0b602e5
·
1 Parent(s): d331050
Files changed (1) hide show
  1. run_job.py +11 -5
run_job.py CHANGED
@@ -55,7 +55,7 @@ def submit_job(
55
  def main():
56
  parser = argparse.ArgumentParser(description="Podcast job runner")
57
  parser.add_argument("--provider", type=str, default="hf-inference")
58
- parser.add_argument("--repo-id", type=str, default="fdaudens/podcast-jobs/podcasts/")
59
  parser.add_argument("--flavor", type=str, default="t4-medium")
60
  args = parser.parse_args()
61
 
@@ -113,11 +113,17 @@ def main():
113
  else:
114
  api = HfApi(token=hf_token)
115
  username = api.whoami()["name"]
116
- repo_id = f"{username}/podcast-jobs/podcasts/" # or your desired repo
117
 
118
- # Create a folder by date
 
 
 
 
 
 
119
  today = datetime.now().strftime("%Y-%m-%d")
120
- remote_path = f"podcast-{today}.wav"
121
 
122
  print(f"Uploading podcast.wav to {repo_id} at {remote_path} ...")
123
  api.upload_file(
@@ -126,7 +132,7 @@ def main():
126
  repo_id=repo_id,
127
  token=hf_token
128
  )
129
- print(f"Uploaded podcast.wav to {repo_id}{remote_path}")
130
  else:
131
  print("No audio generated.")
132
 
 
55
  def main():
56
  parser = argparse.ArgumentParser(description="Podcast job runner")
57
  parser.add_argument("--provider", type=str, default="hf-inference")
58
+ parser.add_argument("--repo-id", type=str, default="fdaudens/podcast-jobs")
59
  parser.add_argument("--flavor", type=str, default="t4-medium")
60
  args = parser.parse_args()
61
 
 
113
  else:
114
  api = HfApi(token=hf_token)
115
  username = api.whoami()["name"]
116
+ repo_id = f"{username}/podcast-jobs" # Correct repo id
117
 
118
+ # Create the repo if it doesn't exist
119
+ try:
120
+ api.create_repo(repo_id, repo_type="model", private=False, exist_ok=True)
121
+ except Exception as e:
122
+ print(f"Warning: Could not create repo (it may already exist): {e}")
123
+
124
+ # Create a folder by date inside the podcasts subfolder
125
  today = datetime.now().strftime("%Y-%m-%d")
126
+ remote_path = f"podcasts/podcast-{today}.wav" # subfolder in repo
127
 
128
  print(f"Uploading podcast.wav to {repo_id} at {remote_path} ...")
129
  api.upload_file(
 
132
  repo_id=repo_id,
133
  token=hf_token
134
  )
135
+ print(f"Uploaded podcast.wav to {repo_id}/{remote_path}")
136
  else:
137
  print("No audio generated.")
138