Spaces:
Sleeping
Sleeping
zhiweili
commited on
Commit
·
82da816
1
Parent(s):
7d76117
fix:resize
Browse files
app.py
CHANGED
@@ -1,5 +1,5 @@
|
|
1 |
import gradio as gr
|
2 |
-
import
|
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 =
|
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,
|