Anish13 commited on
Commit
b729cd2
·
1 Parent(s): 6227aa1
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -178,7 +178,7 @@ model.to(device)
178
  # generate from the model
179
  context = torch.zeros((1, 1), dtype=torch.long, device=device)
180
 
181
- def greet(number_of_tokens, start_character=" "):
182
  context[0][0] = encode(start_character)[0]
183
  max_new_tokens = number_of_tokens
184
  return decode(model.generate(context, max_new_tokens=int(max_new_tokens))[0].tolist())
 
178
  # generate from the model
179
  context = torch.zeros((1, 1), dtype=torch.long, device=device)
180
 
181
+ def greet(number_of_tokens, start_character):
182
  context[0][0] = encode(start_character)[0]
183
  max_new_tokens = number_of_tokens
184
  return decode(model.generate(context, max_new_tokens=int(max_new_tokens))[0].tolist())