Update app.py
Browse files
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 |
-
|
38 |
-
|
39 |
-
|
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 |
|