Spaces:
Running
Running
un-index
commited on
Commit
·
6dcba11
1
Parent(s):
759a131
app.py
CHANGED
|
@@ -66,7 +66,7 @@ def f(context, temperature, top_p, max_length, model_idx):
|
|
| 66 |
#
|
| 67 |
set_seed(randint(1, 999999999999999999))
|
| 68 |
# return sequences specifies how many to return
|
| 69 |
-
generator(context, max_length=max_length, top_p=top_p, temperature=temperature,num_return_sequences=1)
|
| 70 |
# args found in the source: https://github.com/huggingface/transformers/blob/27b3031de2fb8195dec9bc2093e3e70bdb1c4bff/src/transformers/generation_tf_utils.py#L348-L376
|
| 71 |
|
| 72 |
except Exception as e:
|
|
|
|
| 66 |
#
|
| 67 |
set_seed(randint(1, 999999999999999999))
|
| 68 |
# return sequences specifies how many to return
|
| 69 |
+
return generator(context, max_length=max_length, top_p=top_p, temperature=temperature,num_return_sequences=1)
|
| 70 |
# args found in the source: https://github.com/huggingface/transformers/blob/27b3031de2fb8195dec9bc2093e3e70bdb1c4bff/src/transformers/generation_tf_utils.py#L348-L376
|
| 71 |
|
| 72 |
except Exception as e:
|