Update main.py

main.py

@@ -357,7 +357,7 @@ async def websocket_endpoint(websocket: WebSocket):
                 next_frame, next_frame_append = predict_next_frame(previous_frames, previous_actions)
                 # 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")
+                #img = Image.open(f"record_100/image_{82+len(previous_frames)}.png")
                 #previous_frames.append(img)
                 previous_frames.append(next_frame_append)