xunsong.li commited on
Commit
daa45ed
1 Parent(s): 172285b

limit generation params, beautify page

Browse files
Files changed (1) hide show
  1. app.py +20 -15
app.py CHANGED
@@ -31,12 +31,6 @@ snapshot_download(
31
  repo_id="patrolli/AnimateAnyone",
32
  local_dir="./pretrained_weights",
33
  )
34
- is_spaces = True if "SPACE_ID" in os.environ else False
35
- true_for_shared_ui = False # This will be true only if you are in a shared UI
36
- if is_spaces:
37
- true_for_shared_ui = (
38
- True if "xunsong/Moore-AnimateAnyone" in os.environ["SPACE_ID"] else False
39
- )
40
 
41
 
42
  class AnimateController:
@@ -172,9 +166,20 @@ controller = AnimateController()
172
 
173
  def ui():
174
  with gr.Blocks() as demo:
175
- gr.Markdown(
176
  """
177
- # Moore-AnimateAnyone Demo
 
 
 
 
 
 
 
 
 
 
 
178
  """
179
  )
180
  animation = gr.Video(
@@ -192,13 +197,13 @@ def ui():
192
 
193
  with gr.Column():
194
  width_slider = gr.Slider(
195
- label="Width", minimum=448, maximum=768, value=512, step=64
196
  )
197
  height_slider = gr.Slider(
198
- label="Height", minimum=512, maximum=1024, value=768, step=64
199
  )
200
  length_slider = gr.Slider(
201
- label="Video Length", minimum=24, maximum=128, value=24, step=24
202
  )
203
  with gr.Row():
204
  seed_textbox = gr.Textbox(label="Seed", value=-1)
@@ -213,10 +218,10 @@ def ui():
213
  with gr.Row():
214
  sampling_steps = gr.Slider(
215
  label="Sampling steps",
216
- value=25,
217
- info="default: 25",
218
  step=5,
219
- maximum=30,
220
  minimum=10,
221
  )
222
  guidance_scale = gr.Slider(
@@ -224,7 +229,7 @@ def ui():
224
  value=3.5,
225
  info="default: 3.5",
226
  step=0.5,
227
- maximum=10,
228
  minimum=2.0,
229
  )
230
  submit = gr.Button("Animate")
 
31
  repo_id="patrolli/AnimateAnyone",
32
  local_dir="./pretrained_weights",
33
  )
 
 
 
 
 
 
34
 
35
 
36
  class AnimateController:
 
166
 
167
  def ui():
168
  with gr.Blocks() as demo:
169
+ gr.HTML(
170
  """
171
+ <h1 style="color:#dc5b1c;text-align:center">
172
+ Moore-AnimateAnyone Gradio Demo
173
+ </h1>
174
+ <div style="text-align:center">
175
+ <div style="display: inline-block; text-align: left;">
176
+ <p> This is a quick preview demo of Moore-AnimateAnyone. We appreciate the assistance provided by the HuggingFace team in setting up this demo. </p>
177
+
178
+ <p> To reduce waiting time, we limit the size (width, height and length) and inference steps when generating videos. </p>
179
+
180
+ <p> If you like this project, please consider giving a star on <a herf="https://github.com/MooreThreads/Moore-AnimateAnyone"> our GitHub repo </a> 🤗. </p>
181
+ </div>
182
+ </div>
183
  """
184
  )
185
  animation = gr.Video(
 
197
 
198
  with gr.Column():
199
  width_slider = gr.Slider(
200
+ label="Width", minimum=256, maximum=448, value=448, step=64
201
  )
202
  height_slider = gr.Slider(
203
+ label="Height", minimum=256, maximum=512, value=512, step=64
204
  )
205
  length_slider = gr.Slider(
206
+ label="Video Length", minimum=24, maximum=24, value=24, step=1
207
  )
208
  with gr.Row():
209
  seed_textbox = gr.Textbox(label="Seed", value=-1)
 
218
  with gr.Row():
219
  sampling_steps = gr.Slider(
220
  label="Sampling steps",
221
+ value=15,
222
+ info="default: 15",
223
  step=5,
224
+ maximum=15,
225
  minimum=10,
226
  )
227
  guidance_scale = gr.Slider(
 
229
  value=3.5,
230
  info="default: 3.5",
231
  step=0.5,
232
+ maximum=6.5,
233
  minimum=2.0,
234
  )
235
  submit = gr.Button("Animate")