Spaces:
Runtime error
Runtime error
Fix
Browse files
app.py
CHANGED
@@ -3,6 +3,7 @@
|
|
3 |
from __future__ import annotations
|
4 |
|
5 |
import os
|
|
|
6 |
import shlex
|
7 |
import subprocess
|
8 |
|
@@ -27,6 +28,8 @@ pipe = pipeline('text-to-video-synthesis', 'damo/text-to-video-synthesis')
|
|
27 |
|
28 |
|
29 |
def generate(prompt: str, seed: int) -> str:
|
|
|
|
|
30 |
torch.manual_seed(seed)
|
31 |
return pipe({'text': prompt})[OutputKeys.OUTPUT_VIDEO]
|
32 |
|
|
|
3 |
from __future__ import annotations
|
4 |
|
5 |
import os
|
6 |
+
import random
|
7 |
import shlex
|
8 |
import subprocess
|
9 |
|
|
|
28 |
|
29 |
|
30 |
def generate(prompt: str, seed: int) -> str:
|
31 |
+
if seed == -1:
|
32 |
+
seed = random.randint(0, 1000000)
|
33 |
torch.manual_seed(seed)
|
34 |
return pipe({'text': prompt})[OutputKeys.OUTPUT_VIDEO]
|
35 |
|