Update gpt.py
Browse files
gpt.py
CHANGED
@@ -47,13 +47,13 @@ if st.button('Сгенерировать текст'):
|
|
47 |
out = model.generate(
|
48 |
input_ids=prompt,
|
49 |
max_length=length,
|
50 |
-
num_beams=
|
51 |
do_sample=True,
|
52 |
temperature=temperature,
|
53 |
top_k=top_k,
|
54 |
top_p=top_p,
|
55 |
no_repeat_ngram_size=3,
|
56 |
-
num_return_sequences=
|
57 |
).cpu().numpy()
|
58 |
for i, out_ in enumerate(out):
|
59 |
st.write(f'Текст {i+1}:')
|
|
|
47 |
out = model.generate(
|
48 |
input_ids=prompt,
|
49 |
max_length=length,
|
50 |
+
num_beams=5,
|
51 |
do_sample=True,
|
52 |
temperature=temperature,
|
53 |
top_k=top_k,
|
54 |
top_p=top_p,
|
55 |
no_repeat_ngram_size=3,
|
56 |
+
num_return_sequences=num_samples,
|
57 |
).cpu().numpy()
|
58 |
for i, out_ in enumerate(out):
|
59 |
st.write(f'Текст {i+1}:')
|