Update app.py
Browse files
app.py
CHANGED
|
@@ -39,15 +39,15 @@ def update_imgbox(choices):
|
|
| 39 |
def gen_fn(model_str, prompt, neg_prompt=None):
|
| 40 |
if model_str == 'NA':
|
| 41 |
return None
|
|
|
|
| 42 |
if neg_prompt is None:
|
| 43 |
neg_prompt = 'negative prompt'
|
| 44 |
-
|
| 45 |
-
return models_load[model_str](f'{prompt} {noise} {neg_prompt}')
|
| 46 |
|
| 47 |
def make_me():
|
| 48 |
with gr.Row():
|
| 49 |
-
txt_input = gr.Textbox(label='Your prompt:', lines=
|
| 50 |
-
neg_txt_input = gr.Textbox(label='Negative prompt:', lines=
|
| 51 |
|
| 52 |
gen_button = gr.Button('Generate images', width=30, height=30)
|
| 53 |
stop_button = gr.Button('Stop', variant='secondary', interactive=False, width=30, height=30)
|
|
|
|
| 39 |
def gen_fn(model_str, prompt, neg_prompt=None):
|
| 40 |
if model_str == 'NA':
|
| 41 |
return None
|
| 42 |
+
noise = str(randint(0, 9999))
|
| 43 |
if neg_prompt is None:
|
| 44 |
neg_prompt = 'negative prompt'
|
| 45 |
+
return models_load[model_str](prompt=prompt, noise=noise, negative_prompt=neg_prompt)
|
|
|
|
| 46 |
|
| 47 |
def make_me():
|
| 48 |
with gr.Row():
|
| 49 |
+
txt_input = gr.Textbox(label='Your prompt:', lines=2, width=800, max_height=30)
|
| 50 |
+
neg_txt_input = gr.Textbox(label='Negative prompt:', lines=2, width=800, max_height=30)
|
| 51 |
|
| 52 |
gen_button = gr.Button('Generate images', width=30, height=30)
|
| 53 |
stop_button = gr.Button('Stop', variant='secondary', interactive=False, width=30, height=30)
|