Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -43,6 +43,7 @@ def infer(model, replay_file,
|
|
43 |
s = serialize_replay_frame(replay_frame)
|
44 |
serialized.append(s)
|
45 |
serialized = np.stack(serialized)
|
|
|
46 |
it.close()
|
47 |
|
48 |
scoreboard = serialized[:, RF_SCOREBOARD]
|
|
|
43 |
s = serialize_replay_frame(replay_frame)
|
44 |
serialized.append(s)
|
45 |
serialized = np.stack(serialized)
|
46 |
+
serialized = torch.from_numpy(serialized)
|
47 |
it.close()
|
48 |
|
49 |
scoreboard = serialized[:, RF_SCOREBOARD]
|