un-index commited on
Commit
6dcba11
·
1 Parent(s): 759a131
Files changed (1) hide show
  1. app.py +1 -1
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: