pr0mila commited on
Commit
d385ddd
·
1 Parent(s): 09a32cd

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -34,9 +34,9 @@ def get_result_with_bloom(text):
34
  inputs1 = tokenizer(text, return_tensors="pt")
35
  output1 = tokenizer.decode(model.generate(inputs1["input_ids"],
36
  max_length=result_length,
37
- num_beams=2,
38
- no_repeat_ngram_size=2,
39
- early_stopping=True
40
  )[0])
41
  return output1
42
 
 
34
  inputs1 = tokenizer(text, return_tensors="pt")
35
  output1 = tokenizer.decode(model.generate(inputs1["input_ids"],
36
  max_length=result_length,
37
+ do_sample=True,
38
+ top_k=50,
39
+ top_p=0.9
40
  )[0])
41
  return output1
42