prouditalian commited on
Commit
f6b77d7
·
1 Parent(s): 7784478

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -5
app.py CHANGED
@@ -32,13 +32,14 @@ if button:
32
  input_ids = tokenizer.encode(text, return_tensors="pt")
33
  out = model.generate(
34
  input_ids,
35
- min_length=100,
36
- max_length=100,
37
  eos_token_id=5,
38
  #pad_token=1,
39
- top_k=10,
40
- top_p=0.0,
41
- no_repeat_ngram_size=5
 
42
  )
43
  generated_text = list(map(tokenizer.decode, out))[0]
44
  st.subheader("Вот мой ответ:")
 
32
  input_ids = tokenizer.encode(text, return_tensors="pt")
33
  out = model.generate(
34
  input_ids,
35
+ min_length=80,
36
+ max_length=150,
37
  eos_token_id=5,
38
  #pad_token=1,
39
+ do_sample=True,
40
+ top_k=0,
41
+ top_p=0.8,
42
+ no_repeat_ngram_size=4
43
  )
44
  generated_text = list(map(tokenizer.decode, out))[0]
45
  st.subheader("Вот мой ответ:")