Update app.py

app.py

@@ -107,8 +107,6 @@ def generate_video(input_image, motion_prompt, num_frames, guidance, seed, rando
         new_w, new_h = 480, 832
     img = img.resize((new_w, new_h), Image.LANCZOS)
 
-    pipe.to("cuda")
-
     generator = torch.Generator(device="cpu").manual_seed(seed)
 
     output = pipe(
@@ -125,8 +123,6 @@ def generate_video(input_image, motion_prompt, num_frames, guidance, seed, rando
 
     frames = output.frames[0]
 
-    pipe.to("cpu")
-
     tmp = tempfile.NamedTemporaryFile(suffix=".mp4", delete=False)
     export_to_video(frames, tmp.name, fps=16)
 
@@ -339,7 +335,6 @@ with gr.Blocks(css=css, title="VideoGen") as demo:
     </div>
     """)
 
-    # Input image
     input_image = gr.Image(
         label="Input Image",
         type="numpy",
@@ -347,7 +342,6 @@ with gr.Blocks(css=css, title="VideoGen") as demo:
         elem_classes="upload-area",
     )
 
-    # Motion prompt
     gr.HTML('<div class="card"><div class="card-label">✦ Motion — what should move?</div>')
     motion_prompt = gr.Textbox(
         show_label=False,
@@ -356,13 +350,11 @@ with gr.Blocks(css=css, title="VideoGen") as demo:
     )
     gr.HTML('</div>')
 
-    # Generate
     generate_btn = gr.Button(
         "Generate Video ✦", variant="primary",
         size="lg", elem_classes="gen-btn",
     )
 
-    # Output
     output_video = gr.Video(
         label="Generated Video",
         elem_classes="video-out",
@@ -379,7 +371,6 @@ with gr.Blocks(css=css, title="VideoGen") as demo:
         elem_classes="hint-box",
     )
 
-    # Settings
     with gr.Accordion("⚙️  Settings", open=False):
         gr.HTML('<div style="height:6px"></div>')