Spaces:
Runtime error
Runtime error
da03
commited on
Commit
·
21ceee1
1
Parent(s):
a1aa2aa
main.py
CHANGED
|
@@ -221,7 +221,7 @@ def predict_next_frame(previous_frames: List[np.ndarray], previous_actions: List
|
|
| 221 |
|
| 222 |
|
| 223 |
for action_type, pos in previous_actions: #[-8:]:
|
| 224 |
-
print ('here3', action_type, pos)
|
| 225 |
if action_type == 'move':
|
| 226 |
action_type = 'N'
|
| 227 |
if action_type == 'left_click':
|
|
@@ -439,9 +439,9 @@ async def websocket_endpoint(websocket: WebSocket):
|
|
| 439 |
|
| 440 |
action_type = data.get("action_type")
|
| 441 |
mouse_position = data.get("mouse_position")
|
| 442 |
-
if np.random.random() < 0.
|
| 443 |
print ('setting left click')
|
| 444 |
-
action_type = '
|
| 445 |
|
| 446 |
|
| 447 |
# Store the actions
|
|
|
|
| 221 |
|
| 222 |
|
| 223 |
for action_type, pos in previous_actions: #[-8:]:
|
| 224 |
+
#print ('here3', action_type, pos)
|
| 225 |
if action_type == 'move':
|
| 226 |
action_type = 'N'
|
| 227 |
if action_type == 'left_click':
|
|
|
|
| 439 |
|
| 440 |
action_type = data.get("action_type")
|
| 441 |
mouse_position = data.get("mouse_position")
|
| 442 |
+
if np.random.random() < 0.7:
|
| 443 |
print ('setting left click')
|
| 444 |
+
action_type = 'left_click'
|
| 445 |
|
| 446 |
|
| 447 |
# Store the actions
|