Spaces:
Running
Running
Commit
·
413244f
1
Parent(s):
54f4777
Minor
Browse files
app.py
CHANGED
|
@@ -85,11 +85,14 @@ def generate(
|
|
| 85 |
):
|
| 86 |
os.system(f'python gen_t2m.py --gpu_id 0 --seed {seed} --ext {uid} --repeat_times {repeat_times} --motion_length {motion_length} --text_prompt "{text}"')
|
| 87 |
datas = []
|
|
|
|
|
|
|
| 88 |
for n in range(repeat_times):
|
| 89 |
data_unit = {
|
| 90 |
-
"url": f"./generation/{uid}/animations/0/sample0_repeat{n}
|
| 91 |
}
|
| 92 |
datas.append(data_unit)
|
|
|
|
| 93 |
return datas
|
| 94 |
|
| 95 |
|
|
|
|
| 85 |
):
|
| 86 |
os.system(f'python gen_t2m.py --gpu_id 0 --seed {seed} --ext {uid} --repeat_times {repeat_times} --motion_length {motion_length} --text_prompt "{text}"')
|
| 87 |
datas = []
|
| 88 |
+
file_name = [name for name in os.listdir(f"./generation/{uid}/animations/0/") if name.endswith('_ik.mp4')][0]
|
| 89 |
+
motion_length = int(file_name.split('len')[-1].replace('_ik.mp4', ''))
|
| 90 |
for n in range(repeat_times):
|
| 91 |
data_unit = {
|
| 92 |
+
"url": f"./generation/{uid}/animations/0/sample0_repeat{n}_len{motion_length}_ik.mp4"
|
| 93 |
}
|
| 94 |
datas.append(data_unit)
|
| 95 |
+
print(datas)
|
| 96 |
return datas
|
| 97 |
|
| 98 |
|