da03 commited on
Commit
e1579d0
·
1 Parent(s): a9d9852
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
  prompt = " ".join(action_descriptions[-8:])
225
  print(prompt)
226
  #prompt = "N N N N N : N N N N N N N N N N : N N N N N N N N N N : N N N N N N N N N N : N N N N N N N N N N : N N N N N N N N N N : N N N N N N N N N N : N N N N N + 0 3 0 7 : + 0 3 7 5"
227
-
228
- pos_map, leftclick_map, x_scaled, y_scaled = create_position_and_click_map(parse_action_string(action_descriptions[-1]))
229
 
230
 
231
  #prompt = ''
 
224
  prompt = " ".join(action_descriptions[-8:])
225
  print(prompt)
226
  #prompt = "N N N N N : N N N N N N N N N N : N N N N N N N N N N : N N N N N N N N N N : N N N N N N N N N N : N N N N N N N N N N : N N N N N N N N N N : N N N N N + 0 3 0 7 : + 0 3 7 5"
227
+ x, y, action_type = parse_action_string(action_descriptions[-1])
228
+ pos_map, leftclick_map, x_scaled, y_scaled = create_position_and_click_map((x, y), action_type)
229
 
230
 
231
  #prompt = ''