Spaces:
Runtime error
Runtime error
Commit
·
1415561
1
Parent(s):
521709d
Update main.py
Browse files
main.py
CHANGED
@@ -129,7 +129,7 @@ def predict_next_frame(previous_frames: List[np.ndarray], previous_actions: List
|
|
129 |
action_descriptions.append("right_click")
|
130 |
|
131 |
prompt = " ".join(action_descriptions[-8:])
|
132 |
-
prompt = "1~1 0~0 0~0 0~0 0~0 0~0 0~0 0~0"
|
133 |
print(prompt)
|
134 |
|
135 |
# Generate the next frame
|
@@ -154,7 +154,7 @@ async def websocket_endpoint(websocket: WebSocket):
|
|
154 |
previous_frames = []
|
155 |
previous_actions = []
|
156 |
positions = ['815~335', '787~342', '787~342', '749~345', '703~346', '703~346', '654~347', '654~347', '604~349', '555~353', '555~353', '509~357', '509~357', '468~362', '431~368', '431~368']
|
157 |
-
positions = ['815~335', '787~342', '749~345', '703~346', '703~346', '654~347', '654~347', '604~349', '555~353', '555~353', '509~357', '509~357', '468~362', '431~368', '431~368']
|
158 |
|
159 |
#positions = positions[:4]
|
160 |
try:
|
|
|
129 |
action_descriptions.append("right_click")
|
130 |
|
131 |
prompt = " ".join(action_descriptions[-8:])
|
132 |
+
#prompt = "1~1 0~0 0~0 0~0 0~0 0~0 0~0 0~0"
|
133 |
print(prompt)
|
134 |
|
135 |
# Generate the next frame
|
|
|
154 |
previous_frames = []
|
155 |
previous_actions = []
|
156 |
positions = ['815~335', '787~342', '787~342', '749~345', '703~346', '703~346', '654~347', '654~347', '604~349', '555~353', '555~353', '509~357', '509~357', '468~362', '431~368', '431~368']
|
157 |
+
#positions = ['815~335', '787~342', '749~345', '703~346', '703~346', '654~347', '654~347', '604~349', '555~353', '555~353', '509~357', '509~357', '468~362', '431~368', '431~368']
|
158 |
|
159 |
#positions = positions[:4]
|
160 |
try:
|