seawolf2357 commited on
Commit
c687722
·
verified ·
1 Parent(s): 576a49f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +11 -13
app.py CHANGED
@@ -120,17 +120,17 @@ def get_num_frames(duration_seconds: float) -> int:
120
  @spaces.GPU
121
  def generate_video(
122
  prompt: str,
123
- negative_prompt: str = default_negative_prompt,
124
- enhance_prompt_option: bool = False,
125
- duration_seconds: float = 3.0,
126
- guidance_scale: float = 7.5,
127
- num_inference_steps: int = 20,
128
- height: int = 480,
129
- width: int = 832,
130
- seed: int = 0,
131
- randomize_seed: bool = True,
132
- progress=gr.Progress(track_tqdm=True),
133
- ) -> Tuple[str, int, str]:
134
  """Generate video from text prompt."""
135
 
136
  if not prompt.strip():
@@ -710,7 +710,5 @@ with gr.Blocks() as demo:
710
  outputs=outputs,
711
  )
712
 
713
- gr.api(generate_video, api_name="generate_video")
714
-
715
  if __name__ == "__main__":
716
  demo.queue().launch()
 
120
  @spaces.GPU
121
  def generate_video(
122
  prompt: str,
123
+ negative_prompt: str,
124
+ enhance_prompt_option: bool,
125
+ duration_seconds: float,
126
+ guidance_scale: float,
127
+ num_inference_steps: int,
128
+ height: int,
129
+ width: int,
130
+ seed: int,
131
+ randomize_seed: bool,
132
+ progress: gr.Progress = gr.Progress(track_tqdm=True),
133
+ ) -> Tuple[str, int, str, str]:
134
  """Generate video from text prompt."""
135
 
136
  if not prompt.strip():
 
710
  outputs=outputs,
711
  )
712
 
 
 
713
  if __name__ == "__main__":
714
  demo.queue().launch()