Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -18,13 +18,13 @@ def func(sentence, max_length, temperature):
|
|
18 |
output_list = model.generate(
|
19 |
input_ids,
|
20 |
do_sample=True,
|
21 |
-
max_length=max_length,
|
22 |
-
temperature=temperature,
|
23 |
top_k=50,
|
24 |
top_p=0.95,
|
25 |
num_return_sequences=5
|
26 |
)
|
27 |
-
output_strs = [output for output in output_list]
|
28 |
return output_strs
|
29 |
|
30 |
|
|
|
18 |
output_list = model.generate(
|
19 |
input_ids,
|
20 |
do_sample=True,
|
21 |
+
max_length=float(max_length),
|
22 |
+
temperature=float(temperature),
|
23 |
top_k=50,
|
24 |
top_p=0.95,
|
25 |
num_return_sequences=5
|
26 |
)
|
27 |
+
output_strs = [tokenizer.decode(output, skip_special_tokens=True) for output in output_list]
|
28 |
return output_strs
|
29 |
|
30 |
|