Spaces:
Sleeping
Sleeping
Andrei Cozma
commited on
Commit
·
4297e86
1
Parent(s):
58e564b
Updates
Browse files
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=
|
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()
|