Spaces:
Running
Running
Delete Video_generator.py
Browse files- Video_generator.py +0 -14
Video_generator.py
DELETED
@@ -1,14 +0,0 @@
|
|
1 |
-
from diffusers import StableVideoDiffusionPipeline
|
2 |
-
import torch
|
3 |
-
|
4 |
-
# Model Load
|
5 |
-
pipe = StableVideoDiffusionPipeline.from_pretrained(
|
6 |
-
"stabilityai/stable-video-diffusion-img2vid",
|
7 |
-
torch_dtype=torch.float16
|
8 |
-
).to("cuda")
|
9 |
-
|
10 |
-
def generate_video(prompt):
|
11 |
-
video = pipe(prompt, num_inference_steps=25).frames
|
12 |
-
video_path = "generated_video.mp4"
|
13 |
-
video[0].save(video_path) # Save the first generated video
|
14 |
-
return video_path
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|