arr

app.py

@@ -227,6 +227,7 @@ def greet(input):
 	#ctx = F.one_hot(torch.randint(0, 5, (32,)), 5).to(device=device).float()
 
 	# hero, non-hero, food, spell, side-facing
+	one_hot_enc = np.array([1, 0, 0, 0, 0])
 	shape = (32, 5)
 	mtx_2d = np.ones(shape) * one_hot_enc
 	ctx = mtx_2d.to(device=device).float()