Spaces:
Sleeping
Sleeping
ruslanruslanruslan
commited on
Commit
•
60011c0
1
Parent(s):
aa16dec
error fixed
Browse files- 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
|
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])
|