yuntian-deng commited on
Commit
a3217ba
·
1 Parent(s): fff6c83

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +1 -1
main.py CHANGED
@@ -91,7 +91,7 @@ def predict_next_frame(previous_frames: List[np.ndarray], previous_actions: List
91
  # Prepare the prompt based on the previous actions
92
  action_descriptions = []
93
  initial_actions = ['901:604', '901:604', '901:604', '901:604', '901:604', '901:604', '901:604', '921:604']
94
- initial_actions = ['700:897'] * 7
95
  def unnorm_coords(x, y):
96
  return int(x - (1920 - 256) / 2), int(y - (1080 - 256) / 2)
97
 
 
91
  # Prepare the prompt based on the previous actions
92
  action_descriptions = []
93
  initial_actions = ['901:604', '901:604', '901:604', '901:604', '901:604', '901:604', '901:604', '921:604']
94
+ initial_actions = ['0:0'] * 7
95
  def unnorm_coords(x, y):
96
  return int(x - (1920 - 256) / 2), int(y - (1080 - 256) / 2)
97