Spaces:
Running
on
A100
Running
on
A100
Ava Pun
commited on
Commit
·
f75356d
1
Parent(s):
287f91c
Add prompt and seed length limits
Browse files
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.',
|
|
|
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',
|