Rolv-Arild commited on
Commit
f4df420
·
verified ·
1 Parent(s): fa1f0d6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
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]