yuntian-deng commited on
Commit
1415561
·
1 Parent(s): 521709d

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +2 -2
main.py CHANGED
@@ -129,7 +129,7 @@ def predict_next_frame(previous_frames: List[np.ndarray], previous_actions: List
129
  action_descriptions.append("right_click")
130
 
131
  prompt = " ".join(action_descriptions[-8:])
132
- prompt = "1~1 0~0 0~0 0~0 0~0 0~0 0~0 0~0"
133
  print(prompt)
134
 
135
  # Generate the next frame
@@ -154,7 +154,7 @@ async def websocket_endpoint(websocket: WebSocket):
154
  previous_frames = []
155
  previous_actions = []
156
  positions = ['815~335', '787~342', '787~342', '749~345', '703~346', '703~346', '654~347', '654~347', '604~349', '555~353', '555~353', '509~357', '509~357', '468~362', '431~368', '431~368']
157
- positions = ['815~335', '787~342', '749~345', '703~346', '703~346', '654~347', '654~347', '604~349', '555~353', '555~353', '509~357', '509~357', '468~362', '431~368', '431~368']
158
 
159
  #positions = positions[:4]
160
  try:
 
129
  action_descriptions.append("right_click")
130
 
131
  prompt = " ".join(action_descriptions[-8:])
132
+ #prompt = "1~1 0~0 0~0 0~0 0~0 0~0 0~0 0~0"
133
  print(prompt)
134
 
135
  # Generate the next frame
 
154
  previous_frames = []
155
  previous_actions = []
156
  positions = ['815~335', '787~342', '787~342', '749~345', '703~346', '703~346', '654~347', '654~347', '604~349', '555~353', '555~353', '509~357', '509~357', '468~362', '431~368', '431~368']
157
+ #positions = ['815~335', '787~342', '749~345', '703~346', '703~346', '654~347', '654~347', '604~349', '555~353', '555~353', '509~357', '509~357', '468~362', '431~368', '431~368']
158
 
159
  #positions = positions[:4]
160
  try: