Update app.py

app.py

@@ -29,10 +29,11 @@ with open("data.txt", "r", encoding="utf-8") as f:
     chars = sorted(list(set(dataset)))
     vocab_size = len(chars)
 
-stoi = {ch: i for i, ch in enumerate(chars)}  # string to index mapping
-itos = {i: ch for i, ch in enumerate(chars)}  # index to string mapping
-encode = lambda s: [stoi[c] for c in s]
-decode = lambda l: "".join([itos[i] for i in l])
+string_to_int = { ch:i for i,ch in enumerate(chars) }
+int_to_string = { i:ch for i,ch in enumerate(chars) }
+encode = lambda s: [string_to_int[c] for c in s]
+decode = lambda l: ''.join([int_to_string[i] for i in l])
+
 data = torch.tensor(encode(dataset), dtype=torch.long)
 
 class Head(nn.Module):