liruiw commited on
Commit
13dce27
β€’
1 Parent(s): 931641c
Files changed (1) hide show
  1. app.py +5 -5
app.py CHANGED
@@ -14,7 +14,7 @@ genie = GenieSimulator(
14
  quantize=False,
15
  backbone_type='stmar',
16
  backbone_ckpt='data/mar_ckpt/langtable',
17
- prompt_horizon=11,
18
  action_stride=1,
19
  domain='language_table',
20
  )
@@ -64,9 +64,9 @@ if __name__ == '__main__':
64
  right = gr.Button("β†’ Right")
65
 
66
  # Define button interactions
67
- up.click(fn=lambda: handle_input("up"), outputs=image_display)
68
- down.click(fn=lambda: handle_input("down"), outputs=image_display)
69
- left.click(fn=lambda: handle_input("left"), outputs=image_display)
70
- right.click(fn=lambda: handle_input("right"), outputs=image_display)
71
 
72
  demo.launch()
 
14
  quantize=False,
15
  backbone_type='stmar',
16
  backbone_ckpt='data/mar_ckpt/langtable',
17
+ prompt_horizon=5,
18
  action_stride=1,
19
  domain='language_table',
20
  )
 
64
  right = gr.Button("β†’ Right")
65
 
66
  # Define button interactions
67
+ up.click(fn=lambda: handle_input("up"), outputs=image_display, show_progress='hidden')
68
+ down.click(fn=lambda: handle_input("down"), outputs=image_display, show_progress='hidden')
69
+ left.click(fn=lambda: handle_input("left"), outputs=image_display, show_progress='hidden')
70
+ right.click(fn=lambda: handle_input("right"), outputs=image_display, show_progress='hidden')
71
 
72
  demo.launch()