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 |
|