Maximofn commited on
Commit
d8702be
·
1 Parent(s): 512f6a4

Adjust default video generation parameters and GPU decorator

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -65,7 +65,7 @@ def save_video(video_frames, fps=25):
65
  export_to_video(video_frames, tmp_file.name, fps=fps)
66
  return tmp_file.name
67
 
68
- @spaces.GPU(duration=1000)
69
  def generate_video_pipeline(pipeline, prompt, negative_prompt, num_frames, height, width, num_inference_steps):
70
  """Generate video using the pipeline"""
71
  # Move to appropriate device
@@ -140,7 +140,7 @@ def create_interface():
140
  num_frames = gr.Slider(
141
  minimum=8,
142
  maximum=128,
143
- value=50,
144
  step=1,
145
  label="Number of Frames",
146
  info="Number of frames in the video"
@@ -164,7 +164,7 @@ def create_interface():
164
  num_inference_steps = gr.Slider(
165
  minimum=10,
166
  maximum=100,
167
- value=50,
168
  step=1,
169
  label="Inference Steps",
170
  info="Higher number = better quality but slower"
 
65
  export_to_video(video_frames, tmp_file.name, fps=fps)
66
  return tmp_file.name
67
 
68
+ @spaces.GPU()
69
  def generate_video_pipeline(pipeline, prompt, negative_prompt, num_frames, height, width, num_inference_steps):
70
  """Generate video using the pipeline"""
71
  # Move to appropriate device
 
140
  num_frames = gr.Slider(
141
  minimum=8,
142
  maximum=128,
143
+ value=40,
144
  step=1,
145
  label="Number of Frames",
146
  info="Number of frames in the video"
 
164
  num_inference_steps = gr.Slider(
165
  minimum=10,
166
  maximum=100,
167
+ value=40,
168
  step=1,
169
  label="Inference Steps",
170
  info="Higher number = better quality but slower"