yuntian-deng commited on
Commit
dc404d5
·
1 Parent(s): bc57790

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +1 -0
main.py CHANGED
@@ -206,6 +206,7 @@ def predict_next_frame(previous_frames: List[np.ndarray], previous_actions: List
206
  action_descriptions.append("right_click")
207
 
208
  prompt = " ".join(action_descriptions[-8:])
 
209
 
210
  pos_map, x_scaled, y_scaled = create_position_map(parse_action_string(action_descriptions[-1]))
211
 
 
206
  action_descriptions.append("right_click")
207
 
208
  prompt = " ".join(action_descriptions[-8:])
209
+ 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"
210
 
211
  pos_map, x_scaled, y_scaled = create_position_map(parse_action_string(action_descriptions[-1]))
212