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()