Anish13 commited on
Commit
24a5b1e
·
1 Parent(s): f3883fb

TThis should work

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -179,7 +179,7 @@ model.to(device)
179
  context = torch.zeros((1, 1), dtype=torch.long, device=device)
180
 
181
  def greet(start_character, number_of_tokens):
182
- context[0][0] = encode(start_character)
183
  max_new_tokens = number_of_tokens
184
  return decode(model.generate(context, max_new_tokens=max_new_tokens)[0].tolist())
185
 
 
179
  context = torch.zeros((1, 1), dtype=torch.long, device=device)
180
 
181
  def greet(start_character, number_of_tokens):
182
+ context[0][0] = encode(start_character)[0]
183
  max_new_tokens = number_of_tokens
184
  return decode(model.generate(context, max_new_tokens=max_new_tokens)[0].tolist())
185