da03 commited on
Commit
f474968
·
1 Parent(s): f1ec064
Files changed (1) hide show
  1. main.py +1 -1
main.py CHANGED
@@ -507,7 +507,7 @@ async def websocket_endpoint(websocket: WebSocket):
507
  print ('here4', len(previous_frames))
508
  if DEBUG_TEACHER_FORCING:
509
  img = Image.open(f"record_10003/image_{117+len(previous_frames)}.png")
510
- previous_frames.append(img)
511
  else:
512
  #assert False
513
  #previous_frames.append(next_frame_append)
 
507
  print ('here4', len(previous_frames))
508
  if DEBUG_TEACHER_FORCING:
509
  img = Image.open(f"record_10003/image_{117+len(previous_frames)}.png")
510
+ previous_frames.append(np.array(img))
511
  else:
512
  #assert False
513
  #previous_frames.append(next_frame_append)