Spaces:
Running
on
Zero
Running
on
Zero
fix example error
Browse files
app.py
CHANGED
@@ -57,7 +57,7 @@ IMAGE_PROCESS_TRANSFORM = transforms.Compose([
|
|
57 |
])
|
58 |
|
59 |
@spaces.GPU
|
60 |
-
def generate_image(ref_image,
|
61 |
print(f"ref_image: {ref_image.size if ref_image is not None else None}, "
|
62 |
f"ref_image2: {ref_image2.size if ref_image2 is not None else None}, "
|
63 |
f"prompt: {prompt}, height: {height}, width: {width}, num_steps: {num_steps}, guidance_scale: {guidance_scale}, ip_scale: {ip_scale}")
|
@@ -148,7 +148,7 @@ with gr.Blocks() as demo:
|
|
148 |
# When the button is clicked, pass all inputs to generate_image
|
149 |
generate_btn.click(
|
150 |
fn=generate_image,
|
151 |
-
inputs=[ref_image,
|
152 |
outputs=output,
|
153 |
)
|
154 |
|
|
|
57 |
])
|
58 |
|
59 |
@spaces.GPU
|
60 |
+
def generate_image(ref_image, prompt="", height=512, width=512, ref_image2=None, num_steps=25, guidance_scale=3.5, seed=0, ip_scale=1.0):
|
61 |
print(f"ref_image: {ref_image.size if ref_image is not None else None}, "
|
62 |
f"ref_image2: {ref_image2.size if ref_image2 is not None else None}, "
|
63 |
f"prompt: {prompt}, height: {height}, width: {width}, num_steps: {num_steps}, guidance_scale: {guidance_scale}, ip_scale: {ip_scale}")
|
|
|
148 |
# When the button is clicked, pass all inputs to generate_image
|
149 |
generate_btn.click(
|
150 |
fn=generate_image,
|
151 |
+
inputs=[ref_image, description, height_slider, width_slider, ref_image2, steps_slider, guidance_slider, ref_scale_slider],
|
152 |
outputs=output,
|
153 |
)
|
154 |
|