sanjanatule commited on
Commit
dcfc32f
·
1 Parent(s): 58c694a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -18,7 +18,8 @@ decode = lambda l: ''.join([itos[i] for i in l]) # decoder: take a list of integ
18
 
19
 
20
  def generate_dialogue(character_dropdown, seed_slider):
21
-
 
22
  if character_dropdown == "NONE":
23
  context = torch.zeros((1, 1), dtype=torch.long)
24
  return decode(newmodel.model.generate(context, max_new_tokens=100)[0].tolist())
 
18
 
19
 
20
  def generate_dialogue(character_dropdown, seed_slider):
21
+
22
+
23
  if character_dropdown == "NONE":
24
  context = torch.zeros((1, 1), dtype=torch.long)
25
  return decode(newmodel.model.generate(context, max_new_tokens=100)[0].tolist())