da03 commited on
Commit
221d145
·
1 Parent(s): fbec427
Files changed (1) hide show
  1. main.py +2 -2
main.py CHANGED
@@ -224,8 +224,8 @@ def predict_next_frame(previous_frames: List[np.ndarray], previous_actions: List
224
  prev_y = 0
225
  #print ('here')
226
 
227
- prompt = 'N + 0 0 0 0 : + 0 0 4 8 N + 0 2 4 8 : + 0 1 2 0 N + 0 2 4 8 : + 0 3 1 2 N + 0 0 9 6 : + 0 0 1 6 L + 0 4 8 0 : + 0 1 2 0 N + 0 2 9 6 : + 0 0 2 4 N + 0 2 0 0 : + 0 3 0 4 N + 0 2 7 2 : + 0 1 5 2'
228
- previous_actions = [('move', (0, 48)), ('move', (248, 120)), ('move', (248, 312)), ('move', (96, 16)), ('left_click', (480, 120)), ('move', (296, 24)), ('move', (200, 304)), ('move', (272, 152))]
229
 
230
 
231
  for action_type, pos in previous_actions: #[-8:]:
 
224
  prev_y = 0
225
  #print ('here')
226
 
227
+ prompt = 'N + 0 4 1 6 : + 0 3 2 0 L + 0 2 0 0 : + 0 1 7 6 N + 0 3 8 4 : + 0 0 4 8 N + 0 3 6 0 : + 0 2 5 6 N + 0 3 6 8 : + 0 0 1 6 N + 0 0 3 2 : + 0 1 0 4 L + 0 2 8 0 : + 0 0 4 0 L + 0 5 0 4 : + 0 0 7 2'
228
+ previous_actions = [('move', (416, 320)), ('left_click', (200, 176)), ('move', (384, 48)), ('move', (360, 256)), ('move', (368, 16)), ('move', (32, 104)), ('left_click', (280, 40)), ('left_click', (504, 72))]
229
 
230
 
231
  for action_type, pos in previous_actions: #[-8:]: