ChromiumPlutoniumAI commited on
Commit
659a25c
·
verified ·
1 Parent(s): ade3957

Create generate_samples.py

Browse files
Files changed (1) hide show
  1. generate_samples.py +21 -0
generate_samples.py ADDED
@@ -0,0 +1,21 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ from model.multimodal_pipeline import BootyShakerPipeline
2
+
3
+ def showcase_generations():
4
+ pipeline = BootyShakerPipeline()
5
+
6
+ # Text to Video
7
+ txt_prompt = "A person dancing energetically in a club"
8
+ txt_video = pipeline.generate_from_text(txt_prompt)
9
+
10
+ # Image to Video
11
+ img_path = "input_image.jpg"
12
+ img_video = pipeline.generate_from_image(img_path)
13
+
14
+ # Apply different effects
15
+ styles = ["bounce", "wave", "shake"]
16
+ for style in styles:
17
+ altered_video = pipeline.apply_alterations(txt_video, style)
18
+ print(f"Generated {style} video: output_{style}.mp4")
19
+
20
+ if __name__ == "__main__":
21
+ showcase_generations()