Spaces:
Runtime error
Runtime error
jackyliang42
commited on
Commit
·
588af69
1
Parent(s):
47097db
4 blocks, high res
Browse files
app.py
CHANGED
@@ -72,7 +72,7 @@ class DemoRunner:
|
|
72 |
openai.api_key = api_key
|
73 |
self._model_name = model_name
|
74 |
|
75 |
-
self._env = PickPlaceEnv(render=True, high_res=
|
76 |
block_list = np.random.choice(ALL_BLOCKS, size=n_blocks, replace=False).tolist()
|
77 |
bowl_list = np.random.choice(ALL_BOWLS, size=n_bowls, replace=False).tolist()
|
78 |
obj_list = block_list + bowl_list
|
@@ -124,8 +124,8 @@ if __name__ == '__main__':
|
|
124 |
inp_api_key = gr.Textbox(label='OpenAI API Key', lines=1)
|
125 |
inp_model_name = gr.Dropdown(label='Model Name', choices=['code-davinci-002', 'text-davinci-002'], value='code-davinci-002')
|
126 |
with gr.Row():
|
127 |
-
inp_n_blocks = gr.Slider(label='
|
128 |
-
inp_n_bowls = gr.Slider(label='
|
129 |
|
130 |
btn_setup = gr.Button("Setup/Reset Env")
|
131 |
info_setup = gr.Markdown(label='Setup Info')
|
|
|
72 |
openai.api_key = api_key
|
73 |
self._model_name = model_name
|
74 |
|
75 |
+
self._env = PickPlaceEnv(render=True, high_res=True, high_frame_rate=False)
|
76 |
block_list = np.random.choice(ALL_BLOCKS, size=n_blocks, replace=False).tolist()
|
77 |
bowl_list = np.random.choice(ALL_BOWLS, size=n_bowls, replace=False).tolist()
|
78 |
obj_list = block_list + bowl_list
|
|
|
124 |
inp_api_key = gr.Textbox(label='OpenAI API Key', lines=1)
|
125 |
inp_model_name = gr.Dropdown(label='Model Name', choices=['code-davinci-002', 'text-davinci-002'], value='code-davinci-002')
|
126 |
with gr.Row():
|
127 |
+
inp_n_blocks = gr.Slider(label='Number of Blocks', minimum=0, maximum=4, value=3, step=1)
|
128 |
+
inp_n_bowls = gr.Slider(label='Number of Bowls', minimum=0, maximum=4, value=3, step=1)
|
129 |
|
130 |
btn_setup = gr.Button("Setup/Reset Env")
|
131 |
info_setup = gr.Markdown(label='Setup Info')
|