Spaces:
Runtime error
Runtime error
Commit
·
03c7138
1
Parent(s):
7b868a7
Update main.py
Browse files
main.py
CHANGED
@@ -358,7 +358,7 @@ async def websocket_endpoint(websocket: WebSocket):
|
|
358 |
next_frame, next_frame_append = predict_next_frame(previous_frames, previous_actions)
|
359 |
# Load and append the corresponding ground truth image instead of model output
|
360 |
print ('here4', len(previous_frames))
|
361 |
-
if DEBUG_TEACHER_FORCING:
|
362 |
img = Image.open(f"record_100/image_{82+len(previous_frames)}.png")
|
363 |
previous_frames.append(img)
|
364 |
else:
|
|
|
358 |
next_frame, next_frame_append = predict_next_frame(previous_frames, previous_actions)
|
359 |
# Load and append the corresponding ground truth image instead of model output
|
360 |
print ('here4', len(previous_frames))
|
361 |
+
if False and DEBUG_TEACHER_FORCING:
|
362 |
img = Image.open(f"record_100/image_{82+len(previous_frames)}.png")
|
363 |
previous_frames.append(img)
|
364 |
else:
|