Update app.py
Browse files
app.py
CHANGED
@@ -62,7 +62,7 @@ model = load_model()
|
|
62 |
|
63 |
# Text-to-video generation function
|
64 |
@spaces.GPU(duration=120)
|
65 |
-
def generate_video(prompt, image=None, duration=
|
66 |
multiplier = 3
|
67 |
temp = int(duration * multiplier) + 1 # Convert seconds to temp value (assuming 24 FPS)
|
68 |
torch_dtype = torch.bfloat16 if MODEL_DTYPE == "bf16" else torch.float32
|
|
|
62 |
|
63 |
# Text-to-video generation function
|
64 |
@spaces.GPU(duration=120)
|
65 |
+
def generate_video(prompt, image=None, duration=10, guidance_scale=9, video_guidance_scale=5, progress=gr.Progress(track_tqdm=True)):
|
66 |
multiplier = 3
|
67 |
temp = int(duration * multiplier) + 1 # Convert seconds to temp value (assuming 24 FPS)
|
68 |
torch_dtype = torch.bfloat16 if MODEL_DTYPE == "bf16" else torch.float32
|