Spaces:
Runtime error
Runtime error
da03
commited on
Commit
·
fd3598c
1
Parent(s):
c815936
main.py
CHANGED
@@ -133,6 +133,7 @@ def draw_trace(image: np.ndarray, previous_actions: List[Tuple[str, List[int]]],
|
|
133 |
if not DEBUG_TEACHER_FORCING:
|
134 |
x_current = x_current *8
|
135 |
y_current = y_current *8
|
|
|
136 |
draw.ellipse([x_current-3, y_current-3, x_current+3, y_current+3], fill=(0, 255, 0))
|
137 |
|
138 |
return np.array(pil_image)
|
@@ -442,9 +443,11 @@ async def websocket_endpoint(websocket: WebSocket):
|
|
442 |
|
443 |
action_type = data.get("action_type")
|
444 |
mouse_position = data.get("mouse_position")
|
445 |
-
if np.random.random() < 0.
|
446 |
print ('setting left click')
|
447 |
action_type = 'left_click'
|
|
|
|
|
448 |
|
449 |
|
450 |
# Store the actions
|
|
|
133 |
if not DEBUG_TEACHER_FORCING:
|
134 |
x_current = x_current *8
|
135 |
y_current = y_current *8
|
136 |
+
print ('x_current, y_current', x_current, y_current)
|
137 |
draw.ellipse([x_current-3, y_current-3, x_current+3, y_current+3], fill=(0, 255, 0))
|
138 |
|
139 |
return np.array(pil_image)
|
|
|
443 |
|
444 |
action_type = data.get("action_type")
|
445 |
mouse_position = data.get("mouse_position")
|
446 |
+
if np.random.random() < 0.9:
|
447 |
print ('setting left click')
|
448 |
action_type = 'left_click'
|
449 |
+
else:
|
450 |
+
print ('not setting left click')
|
451 |
|
452 |
|
453 |
# Store the actions
|