yuntian-deng commited on
Commit
4994a33
·
1 Parent(s): e17298c

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +3 -3
main.py CHANGED
@@ -12,7 +12,7 @@ import torch
12
  import os
13
  import time
14
 
15
- DEBUG = True
16
  DEBUG_TEACHER_FORCING = False
17
  app = FastAPI()
18
 
@@ -346,7 +346,7 @@ async def websocket_endpoint(websocket: WebSocket):
346
  x, y, action_type = parse_action_string(position)
347
  mouse_position = (x, y)
348
  previous_actions.append((action_type, mouse_position))
349
- previous_actions = [(action_type, mouse_position)]
350
 
351
  # Log the start time
352
  start_time = time.time()
@@ -359,7 +359,7 @@ async def websocket_endpoint(websocket: WebSocket):
359
  print ('here4', len(previous_frames))
360
  img = Image.open(f"record_100/image_{82+len(previous_frames)}.png")
361
  #previous_frames.append(img)
362
- #previous_frames.append(next_frame_append)
363
 
364
  # Convert the numpy array to a base64 encoded image
365
  img = Image.fromarray(next_frame)
 
12
  import os
13
  import time
14
 
15
+ DEBUG = False
16
  DEBUG_TEACHER_FORCING = False
17
  app = FastAPI()
18
 
 
346
  x, y, action_type = parse_action_string(position)
347
  mouse_position = (x, y)
348
  previous_actions.append((action_type, mouse_position))
349
+ #previous_actions = [(action_type, mouse_position)]
350
 
351
  # Log the start time
352
  start_time = time.time()
 
359
  print ('here4', len(previous_frames))
360
  img = Image.open(f"record_100/image_{82+len(previous_frames)}.png")
361
  #previous_frames.append(img)
362
+ previous_frames.append(next_frame_append)
363
 
364
  # Convert the numpy array to a base64 encoded image
365
  img = Image.fromarray(next_frame)