mohamedemam commited on
Commit
098b4c3
1 Parent(s): 106dd33

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -53,7 +53,8 @@ def generate_qa(context, recommended_word, temperature, top_p, num_samples=3):
53
  **input_ids,
54
  temperature=temperature,
55
  top_p=top_p,
56
- num_return_sequences=num_samples
 
57
  )
58
 
59
  generated_text = tokenizer.batch_decode(output, skip_special_tokens=True)
 
53
  **input_ids,
54
  temperature=temperature,
55
  top_p=top_p,
56
+ num_return_sequences=num_samples,do_sample=True, top_k=0
57
+
58
  )
59
 
60
  generated_text = tokenizer.batch_decode(output, skip_special_tokens=True)