AUST001 commited on
Commit
adce73a
·
1 Parent(s): 35b22a1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -234,6 +234,7 @@ demo = gr.Interface(fn=generate_images, inputs=inputs, outputs=outputs,
234
  ["[2, 3, 4]", "Rearrange('c h w -> c w h')"],
235
  ["[2, 3, 4, 4]", "Rearrange('b c h w -> b c (h w)')"],
236
  ["[2, 3, 4, 4]", "Rearrange('b c (h p1) (w p2) -> b (h w) (p1 p2 c)', p1 = 2, p2 = 2)"]
 
237
  ]
238
  )
239
  if __name__ == "__main__":
 
234
  ["[2, 3, 4]", "Rearrange('c h w -> c w h')"],
235
  ["[2, 3, 4, 4]", "Rearrange('b c h w -> b c (h w)')"],
236
  ["[2, 3, 4, 4]", "Rearrange('b c (h p1) (w p2) -> b (h w) (p1 p2 c)', p1 = 2, p2 = 2)"]
237
+ ['[2,3,4,4]', "Rearrange('b c (h p1) (w p2) -> b h w (p1 p2 c)', p1 = 2, p2 = 2)&Rearrange('b h w (c s) -> b w c (h s)', s=2)"]
238
  ]
239
  )
240
  if __name__ == "__main__":