Spaces:
Running
on
Zero
Running
on
Zero
small fix
Browse files
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=
|
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()
|