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
|