zhiweili commited on
Commit
82da816
·
1 Parent(s): 7d76117

fix:resize

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -1,5 +1,5 @@
1
  import gradio as gr
2
- import cv2
3
  import torch
4
  from diffusers import AutoPipelineForImage2Image, DPMSolverMultistepScheduler
5
 
@@ -15,7 +15,7 @@ generator = torch.Generator(device).manual_seed(0)
15
 
16
  def image_to_image(input_image, prompt, guidance_scale, num_inference_steps):
17
  # resize image to 512x512
18
- input_image = cv2.resize(input_image, (512, 512))
19
  # Generate the output image
20
  output_image = pipeline(
21
  generator=generator,
@@ -26,11 +26,11 @@ def image_to_image(input_image, prompt, guidance_scale, num_inference_steps):
26
  return output_image
27
 
28
  with gr.Blocks() as grApp:
29
- input_image = gr.Image(label="Input Image")
30
  prompt = gr.Textbox(lines=3, label="Prompt")
31
  guidance_scale = gr.Slider(minimum=0, maximum=1, value=0.75, label="Guidance Scale")
32
  num_inference_steps = gr.Slider(minimum=10, maximum=100, value=25, label="Number of Inference Steps")
33
- output_image = gr.Image()
34
  generate_btn = gr.Button("Generate Image")
35
  generate_btn.click(
36
  fn=image_to_image,
 
1
  import gradio as gr
2
+ from PIL import Image
3
  import torch
4
  from diffusers import AutoPipelineForImage2Image, DPMSolverMultistepScheduler
5
 
 
15
 
16
  def image_to_image(input_image, prompt, guidance_scale, num_inference_steps):
17
  # resize image to 512x512
18
+ input_image = input_image.resize((512, 512))
19
  # Generate the output image
20
  output_image = pipeline(
21
  generator=generator,
 
26
  return output_image
27
 
28
  with gr.Blocks() as grApp:
29
+ input_image = gr.Image(label="Input Image", type="pil")
30
  prompt = gr.Textbox(lines=3, label="Prompt")
31
  guidance_scale = gr.Slider(minimum=0, maximum=1, value=0.75, label="Guidance Scale")
32
  num_inference_steps = gr.Slider(minimum=10, maximum=100, value=25, label="Number of Inference Steps")
33
+ output_image = gr.Image(label="Output Image", type="pil")
34
  generate_btn = gr.Button("Generate Image")
35
  generate_btn.click(
36
  fn=image_to_image,