broskicodes commited on
Commit
2b34519
·
1 Parent(s): 058eed9

add missing characters in chars list

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -137,7 +137,7 @@ class BigramModel(nn.Module):
137
  # ----------------------------------------------------------------
138
 
139
  # helpers
140
- chars = list("!$&',-.3:;?ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz")
141
  stoi = { ch:i for i,ch in enumerate(chars) }
142
  itos = { i:ch for i,ch in enumerate(chars) }
143
  encode = lambda s: [stoi[c] for c in s] # encoder: take a string, output a list of integers
 
137
  # ----------------------------------------------------------------
138
 
139
  # helpers
140
+ chars = list("\n !$&',-.3:;?ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz")
141
  stoi = { ch:i for i,ch in enumerate(chars) }
142
  itos = { i:ch for i,ch in enumerate(chars) }
143
  encode = lambda s: [stoi[c] for c in s] # encoder: take a string, output a list of integers