alibidaran commited on
Commit
e5cf0cf
·
verified ·
1 Parent(s): e64da6a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -31,7 +31,8 @@ def bot(history):
31
  inputs=tokenizer(text,return_tensors='pt').to('cpu')
32
  with torch.no_grad():
33
  outputs=model.generate(**inputs,max_new_tokens=200,do_sample=True,top_p=0.92,top_k=10,temperature=0.7)
34
- response=tokenizer.decode(outputs[0], skip_special_tokens=True)
 
35
  print(response)
36
  history[-1][1] = ""
37
  for character in response[1:-1]:
 
31
  inputs=tokenizer(text,return_tensors='pt').to('cpu')
32
  with torch.no_grad():
33
  outputs=model.generate(**inputs,max_new_tokens=200,do_sample=True,top_p=0.92,top_k=10,temperature=0.7)
34
+ output_text=outputs[:, inputs.input_ids.shape[1]:]
35
+ response=tokenizer.decode(output_text[0], skip_special_tokens=True)
36
  print(response)
37
  history[-1][1] = ""
38
  for character in response[1:-1]: