Update main.py

main.py

@@ -13,7 +13,7 @@ import os
 import time
 
 DEBUG = True
-DEBUG_TEACHER_FORCING = True
+DEBUG_TEACHER_FORCING = False
 app = FastAPI()
 
 # Mount the static directory to serve HTML, JavaScript, and CSS files
@@ -341,7 +341,7 @@ async def websocket_endpoint(websocket: WebSocket):
                     x, y, action_type = parse_action_string(position)
                     mouse_position = (x, y)
                 previous_actions.append((action_type, mouse_position))
-                #previous_actions = [(action_type, mouse_position)]
+                previous_actions = [(action_type, mouse_position)]
                 
                 # Log the start time
                 start_time = time.time()
@@ -353,7 +353,7 @@ async def websocket_endpoint(websocket: WebSocket):
                 # Load and append the corresponding ground truth image instead of model output
                 print ('here4', len(previous_frames))
                 img = Image.open(f"record_100/image_{82+len(previous_frames)}.png")
-                previous_frames.append(img)
+                #previous_frames.append(img)
                 #previous_frames.append(next_frame_append)
                 
                 # Convert the numpy array to a base64 encoded image