no@email.com commited on
Commit
3c355af
·
1 Parent(s): 9e1aeaa

change access method for dataset

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -21,6 +21,7 @@ import uuid
21
  # =========================================================
22
  MODEL_ID = "Wan-AI/Wan2.2-I2V-A14B-Diffusers"
23
  HF_TOKEN = os.environ.get("HF_TOKEN")
 
24
 
25
  MAX_DIM = 832
26
  MIN_DIM = 480
@@ -263,7 +264,7 @@ with gr.Blocks() as demo:
263
  )
264
  def hf_upload(file_path, prompt, repo):
265
  try:
266
- api=HfApi(token=HF_TOKEN)
267
  unique_name = str(uuid.uuid4())
268
  video_name=f"{unique_name}.mp4"
269
  caption_name=f"{unique_name}.txt"
@@ -283,6 +284,6 @@ def hf_upload(file_path, prompt, repo):
283
  repo_type="dataset"
284
  )
285
  except Exception as e:
286
- print("failed to upload result")
287
  if __name__ == "__main__":
288
  demo.queue().launch(mcp_server=True)
 
21
  # =========================================================
22
  MODEL_ID = "Wan-AI/Wan2.2-I2V-A14B-Diffusers"
23
  HF_TOKEN = os.environ.get("HF_TOKEN")
24
+ DATASET_KEY = os.environ.get("DATASET_KEY")
25
 
26
  MAX_DIM = 832
27
  MIN_DIM = 480
 
264
  )
265
  def hf_upload(file_path, prompt, repo):
266
  try:
267
+ api=HfApi(token=DATASET_KEY)
268
  unique_name = str(uuid.uuid4())
269
  video_name=f"{unique_name}.mp4"
270
  caption_name=f"{unique_name}.txt"
 
284
  repo_type="dataset"
285
  )
286
  except Exception as e:
287
+ print(f"failed to upload result: {e}")
288
  if __name__ == "__main__":
289
  demo.queue().launch(mcp_server=True)