Andrei Cozma commited on
Commit
4297e86
·
1 Parent(s): 58e564b
Files changed (1) hide show
  1. demo.py +4 -1
demo.py CHANGED
@@ -177,6 +177,9 @@ def run(policy_fname, n_test_episodes, max_steps, render_fps, epsilon):
177
  0.0,
178
  1.0,
179
  )
 
 
 
180
 
181
  cv2.putText(
182
  frame_policy,
@@ -394,5 +397,5 @@ with gr.Blocks(title="CS581 Demo") as demo:
394
  ],
395
  )
396
 
397
- demo.queue(concurrency_count=3)
398
  demo.launch()
 
177
  0.0,
178
  1.0,
179
  )
180
+
181
+ # text_color = frame_policy[int((action + 0.5) * frame_policy_res // len(curr_policy)), frame_policy_h // 2]
182
+ # text_color = 1.0 - text_color
183
 
184
  cv2.putText(
185
  frame_policy,
 
397
  ],
398
  )
399
 
400
+ demo.queue(concurrency_count=2)
401
  demo.launch()