debisoft commited on
Commit
b203d27
·
1 Parent(s): f5c772b
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -227,7 +227,7 @@ def greet(input):
227
  #ctx = F.one_hot(torch.randint(0, 5, (32,)), 5).to(device=device).float()
228
 
229
  # hero, non-hero, food, spell, side-facing
230
- shape = (32, 5)
231
  mtx_2d = np.ones(shape) * one_hot_enc
232
  ctx = mtx_2d.to(device=device).float()
233
 
 
227
  #ctx = F.one_hot(torch.randint(0, 5, (32,)), 5).to(device=device).float()
228
 
229
  # hero, non-hero, food, spell, side-facing
230
+ shape = (32, 5)
231
  mtx_2d = np.ones(shape) * one_hot_enc
232
  ctx = mtx_2d.to(device=device).float()
233