matteomarjanovic commited on
Commit
b3b3df1
·
1 Parent(s): 19d8964

fix image width and height

Browse files
Files changed (1) hide show
  1. app.py +3 -2
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=1420,
130
- height=1080,
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