ldhldh commited on
Commit
d7136d1
ยท
1 Parent(s): a2f0814

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -75,7 +75,7 @@ def chat(id, npc, prompt):
75
  inputs = tokenizer(prom, return_tensors="pt")["input_ids"]
76
  outputs = model.generate(inputs, max_new_tokens=100)
77
  print(tokenizer.decode(outputs[0]))
78
- output = outputs[len(prom):-1]
79
  print(output)
80
  else:
81
  output = "no model"
 
75
  inputs = tokenizer(prom, return_tensors="pt")["input_ids"]
76
  outputs = model.generate(inputs, max_new_tokens=100)
77
  print(tokenizer.decode(outputs[0]))
78
+ output = tokenizer.decode(outputs[0])[len(prom):-1]
79
  print(output)
80
  else:
81
  output = "no model"