Spaces:
Running
on
Zero
Running
on
Zero
no@email.com
commited on
Commit
·
3c355af
1
Parent(s):
9e1aeaa
change access method for dataset
Browse files
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=
|
| 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)
|