Nick088 commited on
Commit
e857352
·
verified ·
1 Parent(s): 7eddf91

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -16,6 +16,7 @@ model = T5ForConditionalGeneration.from_pretrained("roborovski/superprompt-v1",
16
  model.to(device)
17
 
18
  def generate(your_prompt, task_prefix, max_new_tokens, repetition_penalty, temperature, model_precision_type, top_p, top_k, seed):
 
19
  if seed == 0:
20
  seed = random.randint(1, 2**32-1)
21
  transformers.set_seed(seed)
@@ -60,7 +61,7 @@ top_p = gr.Slider(value=1, minimum=0, maximum=2, step=0.05, label="Top P", info=
60
 
61
  top_k = gr.Slider(value=50, minimum=1, maximum=100, step=1, label="Top K", info="Higher k means more diverse outputs by considering a range of tokens")
62
 
63
- seed = gr.Slider(value=42, minimum=0, maximum=2**32-1, label="Seed", info="A starting point to initiate the generation process, put 0 for a random one")
64
 
65
  examples = [
66
  ["A storefront with 'Text to Image' written on it.", 512, 1.2, 0.5, "fp16", 1, 50, 42]
 
16
  model.to(device)
17
 
18
  def generate(your_prompt, task_prefix, max_new_tokens, repetition_penalty, temperature, model_precision_type, top_p, top_k, seed):
19
+
20
  if seed == 0:
21
  seed = random.randint(1, 2**32-1)
22
  transformers.set_seed(seed)
 
61
 
62
  top_k = gr.Slider(value=50, minimum=1, maximum=100, step=1, label="Top K", info="Higher k means more diverse outputs by considering a range of tokens")
63
 
64
+ seed = gr.Slider(value=42, minimum=0, maximum=2**32-1, step=1, label="Seed", info="A starting point to initiate the generation process, put 0 for a random one")
65
 
66
  examples = [
67
  ["A storefront with 'Text to Image' written on it.", 512, 1.2, 0.5, "fp16", 1, 50, 42]