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)
|