.

main.py

@@ -512,6 +512,7 @@ async def websocket_endpoint(websocket: WebSocket):
                 #    print ('predicting', f"record_10003/image_{117+len(previous_frames)}.png")
                 print ('previous_actions', previous_actions)
                 next_frame, next_frame_append = predict_next_frame(previous_frames, previous_actions)
+                previous_frames.append(next_frame_append)
                 # Load and append the corresponding ground truth image instead of model output
                 #print ('here4', len(previous_frames))
                 #if DEBUG_TEACHER_FORCING:
@@ -521,8 +522,8 @@ async def websocket_endpoint(websocket: WebSocket):
                 #    assert False
                 #    previous_frames.append(next_frame_append)
                 #    pass
-                previous_frames = []
-                previous_actions = []
+                #previous_frames = []
+                #previous_actions = []
                 
                 # Convert the numpy array to a base64 encoded image
                 img = Image.fromarray(next_frame)