roman-talker commited on
Commit
8403d4a
·
verified ·
1 Parent(s): 175df45

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +16 -19
app.py CHANGED
@@ -1,28 +1,26 @@
1
  import gradio as gr
2
- import torch
3
- from diffusers import WanPipeline, UniPCMultistepScheduler
4
-
5
- pipe = WanPipeline.from_pretrained(
6
- "Wan-AI/Wan2.2-TI2V-5B-Diffusers",
7
- torch_dtype=torch.float16
8
- )
9
- pipe.scheduler = UniPCMultistepScheduler.from_config(pipe.scheduler.config)
10
- pipe = pipe.to("cuda")
11
 
12
  def generate_video(prompt, duration):
13
  """
14
- Generates a video from a text prompt locally using Wan 2.2.
15
- Duration is clamped between 1 and 8 seconds.
16
- Returns the path to the saved MP4 file.
17
  """
18
  duration = max(1, min(duration, 8))
19
- video = pipe(prompt=prompt, height=512, width=512, duration_seconds=duration, guidance_scale=1.0).videos[0]
20
- video_path = "output.mp4"
21
- video.save(video_path)
22
- return video_path
 
 
 
 
 
23
 
 
24
  with gr.Blocks() as demo:
25
- gr.Markdown("## Wan 2.2 Video Generator (Local)")
26
 
27
  with gr.Row():
28
  prompt_input = gr.Textbox(label="Prompt", placeholder="Describe your scene")
@@ -32,5 +30,4 @@ with gr.Blocks() as demo:
32
 
33
  submit_btn.click(generate_video, inputs=[prompt_input, duration_input], outputs=output_video)
34
 
35
-
36
- demo.launch(api_name="generate-video")
 
1
  import gradio as gr
2
+ import time
3
+ import shutil
 
 
 
 
 
 
 
4
 
5
  def generate_video(prompt, duration):
6
  """
7
+ Simulates video generation for testing UI and API.
8
+ Returns a placeholder MP4 file.
 
9
  """
10
  duration = max(1, min(duration, 8))
11
+ print(f"Generating video for prompt: '{prompt}' with duration {duration}s")
12
+
13
+ time.sleep(1)
14
+
15
+ placeholder_video = "placeholder.mp4"
16
+ output_path = "output.mp4"
17
+ shutil.copyfile(placeholder_video, output_path)
18
+
19
+ return output_path
20
 
21
+ # Gradio UI + API
22
  with gr.Blocks() as demo:
23
+ gr.Markdown("## Wan 2.2 Video Generator (Local Test Placeholder)")
24
 
25
  with gr.Row():
26
  prompt_input = gr.Textbox(label="Prompt", placeholder="Describe your scene")
 
30
 
31
  submit_btn.click(generate_video, inputs=[prompt_input, duration_input], outputs=output_video)
32
 
33
+ demo.launch(api_name="generate-video")