Spaces:
Sleeping
Sleeping
Commit
·
098b4c3
1
Parent(s):
106dd33
Update app.py
Browse files
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 |
-
|
|
|
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)
|