Faizbulbul commited on
Commit
007653c
·
verified ·
1 Parent(s): 575b8ce

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +14 -16
app.py CHANGED
@@ -1,20 +1,18 @@
1
- from diffusers import StableVideoDiffusionPipeline
2
  import torch
3
- from PIL import Image
 
 
 
 
4
 
5
- # Model Load
6
- pipe = StableVideoDiffusionPipeline.from_pretrained(
7
- "stabilityai/stable-video-diffusion-img2vid",
8
- torch_dtype=torch.float16
9
- ).to("cuda")
10
 
11
- def generate_video(image_path, prompt):
12
- image = Image.open(image_path).convert("RGB") # Load image properly
13
- video = pipe(prompt=prompt, image=image, num_inference_steps=25).frames
14
- video_path = "generated_video.mp4"
15
- video[0].save(video_path) # Save the first generated video
16
- return video_path
17
 
18
- # Example Usage
19
- video_file = generate_video("sample_image.jpg", "A beautiful mountain landscape with a sunset")
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 पर वीडियो सेव होगा