jackyliang42 commited on
Commit
588af69
·
1 Parent(s): 47097db

4 blocks, high res

Browse files
Files changed (1) hide show
  1. app.py +3 -3
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=False, 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,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='Num Blocks', minimum=0, maximum=3, value=3, step=1)
128
- inp_n_bowls = gr.Slider(label='Num Bowls', minimum=0, maximum=3, value=3, step=1)
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')