ruslanruslanruslan commited on
Commit
60011c0
1 Parent(s): aa16dec

error fixed

Browse files
Files changed (1) hide show
  1. pages/Borgesian.py +1 -1
pages/Borgesian.py CHANGED
@@ -17,7 +17,7 @@ borgesian.eval()
17
  def generate_response(text, temperature, length, top_p):
18
  input_ids = tokenizer.encode(text, return_tensors="pt")
19
  with torch.no_grad():
20
- out = borgesian.generate(input_ids, do_sample=True, num_beams=2, temperature=float(temperature), top_p=float(top_p), max_length=length, truncate=".")
21
  generated_text = list(map(tokenizer.decode, out))[0]
22
  last_full_stop_index = generated_text.rfind('.')
23
  st.write(generated_text[:last_full_stop_index + 1])
 
17
  def generate_response(text, temperature, length, top_p):
18
  input_ids = tokenizer.encode(text, return_tensors="pt")
19
  with torch.no_grad():
20
+ out = borgesian.generate(input_ids, do_sample=True, num_beams=2, temperature=float(temperature), top_p=float(top_p), max_length=length)
21
  generated_text = list(map(tokenizer.decode, out))[0]
22
  last_full_stop_index = generated_text.rfind('.')
23
  st.write(generated_text[:last_full_stop_index + 1])