jiuface commited on
Commit
527d179
·
verified ·
1 Parent(s): 3594837

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -122,7 +122,7 @@ def process(image, image_url, prompt, n_prompt, num_steps, guidance_scale, contr
122
 
123
  size = (orginal_image.size[0], orginal_image.size[1])
124
  print("gorinal image size", size)
125
- depth_image = get_depth_map(orginal_image, progress)
126
  generator = torch.Generator().manual_seed(seed)
127
  print(prompt, n_prompt, guidance_scale, num_steps, control_strength)
128
  print("run pipe")
@@ -139,7 +139,7 @@ def process(image, image_url, prompt, n_prompt, num_steps, guidance_scale, contr
139
  ).images[0]
140
  print("geneate image success")
141
  if upload_to_s3:
142
- url = upload_image_to_s3(generated_image, account_id, access_key, secret_key, bucket, progress)
143
  result = {"status": "success", "url": url}
144
  else:
145
  result = {"status": "success", "message": "Image generated but not uploaded"}
 
122
 
123
  size = (orginal_image.size[0], orginal_image.size[1])
124
  print("gorinal image size", size)
125
+ depth_image = get_depth_map(orginal_image)
126
  generator = torch.Generator().manual_seed(seed)
127
  print(prompt, n_prompt, guidance_scale, num_steps, control_strength)
128
  print("run pipe")
 
139
  ).images[0]
140
  print("geneate image success")
141
  if upload_to_s3:
142
+ url = upload_image_to_s3(generated_image, account_id, access_key, secret_key, bucket)
143
  result = {"status": "success", "url": url}
144
  else:
145
  result = {"status": "success", "message": "Image generated but not uploaded"}