Spaces:
Runtime error
Runtime error
Commit
·
b3b3df1
1
Parent(s):
19d8964
fix image width and height
Browse files
app.py
CHANGED
@@ -120,14 +120,15 @@ def generate_description_fn(
|
|
120 |
detect_resolution=1024,
|
121 |
image_resolution=1024
|
122 |
)
|
|
|
123 |
|
124 |
image = pipe(
|
125 |
prompt=prompt,
|
126 |
control_image=control_image,
|
127 |
guidance_scale=0.,
|
128 |
num_inference_steps=4,
|
129 |
-
width=
|
130 |
-
height=
|
131 |
max_sequence_length=256,
|
132 |
).images[0]
|
133 |
|
|
|
120 |
detect_resolution=1024,
|
121 |
image_resolution=1024
|
122 |
)
|
123 |
+
width, height = control_image.size
|
124 |
|
125 |
image = pipe(
|
126 |
prompt=prompt,
|
127 |
control_image=control_image,
|
128 |
guidance_scale=0.,
|
129 |
num_inference_steps=4,
|
130 |
+
width=width,
|
131 |
+
height=height,
|
132 |
max_sequence_length=256,
|
133 |
).images[0]
|
134 |
|