AngeT10 commited on
Commit
bc05d99
1 Parent(s): e5b94cf

Create app.py

Browse files
Files changed (1) hide show
  1. app.py +25 -0
app.py ADDED
@@ -0,0 +1,25 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import torch
2
+ from diffusers import MochiPipeline
3
+ import gradio as gr
4
+
5
+ # Carica il modello MochiPipeline
6
+ pipe = MochiPipeline.from_pretrained("genmo/mochi-1-preview")
7
+
8
+ # Funzione per generare video (usa meno frame per risparmiare memoria)
9
+ def generate_video(prompt):
10
+ with torch.autocast("cuda", torch.float16):
11
+ frames = pipe(prompt, num_frames=30).frames # Riduci i frame a 30
12
+ frames[0].save("mochi.mp4", format="mp4", save_all=True, duration=100)
13
+ return "mochi.mp4"
14
+
15
+ # Configura l'interfaccia Gradio
16
+ interface = gr.Interface(
17
+ fn=generate_video,
18
+ inputs="text",
19
+ outputs="video",
20
+ title="Video Generator",
21
+ description="Genera un video breve basato sul prompt fornito."
22
+ )
23
+
24
+ # Avvia l'interfaccia
25
+ interface.launch()