Spaces:
Runtime error
Runtime error
Commit
·
4994a33
1
Parent(s):
e17298c
Update main.py
Browse files
main.py
CHANGED
@@ -12,7 +12,7 @@ import torch
|
|
12 |
import os
|
13 |
import time
|
14 |
|
15 |
-
DEBUG =
|
16 |
DEBUG_TEACHER_FORCING = False
|
17 |
app = FastAPI()
|
18 |
|
@@ -346,7 +346,7 @@ async def websocket_endpoint(websocket: WebSocket):
|
|
346 |
x, y, action_type = parse_action_string(position)
|
347 |
mouse_position = (x, y)
|
348 |
previous_actions.append((action_type, mouse_position))
|
349 |
-
previous_actions = [(action_type, mouse_position)]
|
350 |
|
351 |
# Log the start time
|
352 |
start_time = time.time()
|
@@ -359,7 +359,7 @@ async def websocket_endpoint(websocket: WebSocket):
|
|
359 |
print ('here4', len(previous_frames))
|
360 |
img = Image.open(f"record_100/image_{82+len(previous_frames)}.png")
|
361 |
#previous_frames.append(img)
|
362 |
-
|
363 |
|
364 |
# Convert the numpy array to a base64 encoded image
|
365 |
img = Image.fromarray(next_frame)
|
|
|
12 |
import os
|
13 |
import time
|
14 |
|
15 |
+
DEBUG = False
|
16 |
DEBUG_TEACHER_FORCING = False
|
17 |
app = FastAPI()
|
18 |
|
|
|
346 |
x, y, action_type = parse_action_string(position)
|
347 |
mouse_position = (x, y)
|
348 |
previous_actions.append((action_type, mouse_position))
|
349 |
+
#previous_actions = [(action_type, mouse_position)]
|
350 |
|
351 |
# Log the start time
|
352 |
start_time = time.time()
|
|
|
359 |
print ('here4', len(previous_frames))
|
360 |
img = Image.open(f"record_100/image_{82+len(previous_frames)}.png")
|
361 |
#previous_frames.append(img)
|
362 |
+
previous_frames.append(next_frame_append)
|
363 |
|
364 |
# Convert the numpy array to a base64 encoded image
|
365 |
img = Image.fromarray(next_frame)
|