BootyShakerAI / generate_samples.py
ChromiumPlutoniumAI's picture
Create generate_samples.py
659a25c verified
raw
history blame contribute delete
666 Bytes
from model.multimodal_pipeline import BootyShakerPipeline
def showcase_generations():
pipeline = BootyShakerPipeline()
# Text to Video
txt_prompt = "A person dancing energetically in a club"
txt_video = pipeline.generate_from_text(txt_prompt)
# Image to Video
img_path = "input_image.jpg"
img_video = pipeline.generate_from_image(img_path)
# Apply different effects
styles = ["bounce", "wave", "shake"]
for style in styles:
altered_video = pipeline.apply_alterations(txt_video, style)
print(f"Generated {style} video: output_{style}.mp4")
if __name__ == "__main__":
showcase_generations()