dreamdrop-art commited on
Commit
d31b23b
1 Parent(s): 6f838ab

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -0
app.py CHANGED
@@ -361,17 +361,21 @@ with gr.Blocks(css=css) as demo:
361
  with gr.Row():
362
  with gr.Column():
363
  image_input = gr.Image(type="pil")
 
364
 
365
  with gr.Column():
366
  exif_output = gr.HTML(label="EXIF Data")
367
  send_to_txt2img_btn = gr.Button("Send to txt2img")
 
368
 
369
  image_input.upload(get_exif_data, inputs=[image_input], outputs=exif_output)
 
370
  send_to_txt2img_btn.click(send_to_txt2img, inputs=[image_input], outputs=[tabs, prompt, negative_prompt,
371
  steps, seed, model, sampler,
372
  width, height, cfg_scale],
373
  concurrency_limit=64)
374
  send_to_png.click(send_to_img2img_def, inputs=image_output, outputs=image_input)
 
375
  send_to_png_i2i.click(send_to_img2img_def, inputs=i2i_image_output, outputs=image_input)
376
  with gr.Tab("HuggingFace Inference"):
377
  with gr.Row():
 
361
  with gr.Row():
362
  with gr.Column():
363
  image_input = gr.Image(type="pil")
364
+ png_button = gr.Button("Get Info")
365
 
366
  with gr.Column():
367
  exif_output = gr.HTML(label="EXIF Data")
368
  send_to_txt2img_btn = gr.Button("Send to txt2img")
369
+ send_to_img2img_png = gr.Button("Send to img2img")
370
 
371
  image_input.upload(get_exif_data, inputs=[image_input], outputs=exif_output)
372
+ png_button.click(get_exif_data, inputs=[image_input], outputs=exif_output)
373
  send_to_txt2img_btn.click(send_to_txt2img, inputs=[image_input], outputs=[tabs, prompt, negative_prompt,
374
  steps, seed, model, sampler,
375
  width, height, cfg_scale],
376
  concurrency_limit=64)
377
  send_to_png.click(send_to_img2img_def, inputs=image_output, outputs=image_input)
378
+ send_to_img2img_png.click(send_to_img2img_def, inputs=image_input, outputs=i2i_image_input)
379
  send_to_png_i2i.click(send_to_img2img_def, inputs=i2i_image_output, outputs=image_input)
380
  with gr.Tab("HuggingFace Inference"):
381
  with gr.Row():