Spaces:
Runtime error
Runtime error
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')
|