wanghaofan commited on
Commit
ed8566f
·
verified ·
1 Parent(s): bf3b3f7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -126,7 +126,7 @@ def run_lora(input_image, prompt, cfg_scale, steps, selected_index, randomize_se
126
  guidance_scale=cfg_scale,
127
  generator=generator,
128
  joint_attention_kwargs={"scale": lora_scale},
129
- )
130
 
131
  # # Consume the generator to get the final image
132
  # final_image = None
@@ -162,7 +162,7 @@ with gr.Blocks(theme=gr.themes.Soft(), css=css) as app:
162
  selected_index = gr.State(None)
163
  with gr.Row():
164
  with gr.Column(scale=3):
165
- prompt = gr.Textbox(label="Prompt", lines=1, placeholder="Type a prompt after selecting a LoRA")
166
  with gr.Column(scale=1, elem_id="gen_column"):
167
  generate_button = gr.Button("Generate", variant="primary", elem_id="gen_btn")
168
  with gr.Row():
@@ -181,7 +181,7 @@ with gr.Blocks(theme=gr.themes.Soft(), css=css) as app:
181
 
182
  with gr.Column():
183
  progress_bar = gr.Markdown(elem_id="progress",visible=False)
184
- result = gr.Image(label="Generated Image")
185
 
186
  with gr.Row():
187
  with gr.Accordion("Advanced Settings", open=False):
 
126
  guidance_scale=cfg_scale,
127
  generator=generator,
128
  joint_attention_kwargs={"scale": lora_scale},
129
+ ).images[0]
130
 
131
  # # Consume the generator to get the final image
132
  # final_image = None
 
162
  selected_index = gr.State(None)
163
  with gr.Row():
164
  with gr.Column(scale=3):
165
+ prompt = gr.Textbox(label="Prompt", lines=1, placeholder="Please select a LoRA by clicking")
166
  with gr.Column(scale=1, elem_id="gen_column"):
167
  generate_button = gr.Button("Generate", variant="primary", elem_id="gen_btn")
168
  with gr.Row():
 
181
 
182
  with gr.Column():
183
  progress_bar = gr.Markdown(elem_id="progress",visible=False)
184
+ result = gr.Image(label="Generated Image",show_label=False,interactive=False))
185
 
186
  with gr.Row():
187
  with gr.Accordion("Advanced Settings", open=False):