aatir commited on
Commit
c2cf3a4
·
1 Parent(s): 4dcbef1

fixed diffusion fn

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -195,7 +195,7 @@ def preprocess_product_image(image, target_width, target_height):
195
  @spaces.GPU
196
  @torch.inference_mode()
197
  def diffusion_fn(chatbot, canvas_outputs, num_samples, seed, image_width, image_height,
198
- highres_scale, steps, cfg, highres_steps, highres_denoise, negative_prompt):
199
 
200
  use_initial_latent = False
201
  eps = 0.05
 
195
  @spaces.GPU
196
  @torch.inference_mode()
197
  def diffusion_fn(chatbot, canvas_outputs, num_samples, seed, image_width, image_height,
198
+ highres_scale, steps, cfg, highres_steps, highres_denoise, negative_prompt, product_image):
199
 
200
  use_initial_latent = False
201
  eps = 0.05