Spaces:
Runtime error
Runtime error
da03
commited on
Commit
·
292dd30
1
Parent(s):
d0185ca
main.py
CHANGED
@@ -383,7 +383,7 @@ async def websocket_endpoint(websocket: WebSocket):
|
|
383 |
next_frame, next_frame_append = predict_next_frame(previous_frames, previous_actions)
|
384 |
# Load and append the corresponding ground truth image instead of model output
|
385 |
print ('here4', len(previous_frames))
|
386 |
-
if
|
387 |
img = Image.open(f"record_100/image_{82+len(previous_frames)}.png")
|
388 |
previous_frames.append(img)
|
389 |
elif False:
|
|
|
383 |
next_frame, next_frame_append = predict_next_frame(previous_frames, previous_actions)
|
384 |
# Load and append the corresponding ground truth image instead of model output
|
385 |
print ('here4', len(previous_frames))
|
386 |
+
if False and DEBUG_TEACHER_FORCING:
|
387 |
img = Image.open(f"record_100/image_{82+len(previous_frames)}.png")
|
388 |
previous_frames.append(img)
|
389 |
elif False:
|