Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -1,20 +1,18 @@
|
|
1 |
-
from diffusers import StableVideoDiffusionPipeline
|
2 |
import torch
|
3 |
-
from
|
|
|
|
|
|
|
|
|
4 |
|
5 |
-
#
|
6 |
-
|
7 |
-
|
8 |
-
|
9 |
-
).to("cuda")
|
10 |
|
11 |
-
|
12 |
-
|
13 |
-
|
14 |
-
video_path = "generated_video.mp4"
|
15 |
-
video[0].save(video_path) # Save the first generated video
|
16 |
-
return video_path
|
17 |
|
18 |
-
#
|
19 |
-
|
20 |
-
print(f"Video generated and saved at: {video_file}")
|
|
|
|
|
1 |
import torch
|
2 |
+
from diffusers import StableVideoDiffusionPipeline
|
3 |
+
|
4 |
+
# CPU पर लोड करने के लिए
|
5 |
+
pipeline = StableVideoDiffusionPipeline.from_pretrained("stabilityai/stable-video-diffusion-img2vid")
|
6 |
+
pipeline.to("cpu", torch.float32) # ✅ GPU की जगह CPU का इस्तेमाल
|
7 |
|
8 |
+
# इनफरेंस फंक्शन (वीडियो जनरेट करना)
|
9 |
+
def generate_video(prompt):
|
10 |
+
video = pipeline(prompt).videos # ✅ अब यह CPU पर रन करेगा
|
11 |
+
return video
|
|
|
12 |
|
13 |
+
# टेस्टिंग
|
14 |
+
prompt = "A cute cat playing with a ball"
|
15 |
+
output_video = generate_video(prompt)
|
|
|
|
|
|
|
16 |
|
17 |
+
# आउटपुट सेव करें
|
18 |
+
output_video[0].save("generated_video.mp4") # ✅ CPU पर वीडियो सेव होगा
|
|