Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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 |
-
|
|
|
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]:
|