Spaces:
Running
Running
File size: 863 Bytes
c8d7040 72093e1 c8d7040 72093e1 c8d7040 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 |
import torch
import gradio as gr
from diffusers import DiffusionPipeline
from diffusers.utils import export_to_video
# Charger le modèle de diffusion sans utiliser CUDA
pipe = DiffusionPipeline.from_pretrained("damo-vilab/text-to-video-ms-1.7b", torch_dtype=torch.float16, variant="fp16")
pipe = pipe.to("cpu") # Utiliser le CPU au lieu du GPU
# Fonction pour générer la vidéo
def generate_video(prompt):
video_frames = pipe(prompt).frames[0]
video_path = export_to_video(video_frames)
return video_path
# Interface Gradio
iface = gr.Interface(
fn=generate_video,
inputs=gr.Textbox(label="Enter Prompt", placeholder="e.g., Spiderman is surfing"),
outputs=gr.Video(label="Generated Video"),
title="Text-to-Video Generation",
description="Generate a video based on a textual prompt."
)
# Lancer l'interface
iface.launch()
|