Ava Pun commited on
Commit
f75356d
·
1 Parent(s): 287f91c

Add prompt and seed length limits

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -44,10 +44,11 @@ def main():
44
  generator = LegoGenerator(LegoGPT(model_cfg))
45
 
46
  # Define inputs and outputs
47
- in_prompt = gr.Textbox(label='Prompt', placeholder='Enter a prompt to generate a LEGO model.')
48
  in_temperature = gr.Slider(0.01, 2.0, value=model_cfg.temperature, step=0.01,
49
  label='Temperature', info=get_help_string('temperature'))
50
- in_seed = gr.Number(value=42, label='Seed', info='Random seed for generation.', precision=0, step=1)
 
51
  in_bricks = gr.Number(value=model_cfg.max_bricks, label='Max bricks', info=get_help_string('max_bricks'),
52
  precision=0, minimum=1, step=1)
53
  in_rejections = gr.Number(value=model_cfg.max_brick_rejections, label='Max brick rejections',
 
44
  generator = LegoGenerator(LegoGPT(model_cfg))
45
 
46
  # Define inputs and outputs
47
+ in_prompt = gr.Textbox(label='Prompt', placeholder='Enter a prompt to generate a LEGO model.', max_length=500)
48
  in_temperature = gr.Slider(0.01, 2.0, value=model_cfg.temperature, step=0.01,
49
  label='Temperature', info=get_help_string('temperature'))
50
+ in_seed = gr.Number(value=42, label='Seed', info='Random seed for generation.',
51
+ precision=0, minimum=0, maximum=2 ** 32 - 1, step=1)
52
  in_bricks = gr.Number(value=model_cfg.max_bricks, label='Max bricks', info=get_help_string('max_bricks'),
53
  precision=0, minimum=1, step=1)
54
  in_rejections = gr.Number(value=model_cfg.max_brick_rejections, label='Max brick rejections',