chips commited on
Commit
f2e5577
·
1 Parent(s): faf3fab

more on variations

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -48,8 +48,8 @@ async def virtual_try_on(
48
  garment_image_3: UploadFile = File(None),
49
  garment_image_4: UploadFile = File(None),):
50
  """
51
- Pass in the top garment image, bottom garment image, top back garment image, bottom back garment image, talent trigger word, talent lora url and number of images to generate.
52
- Only one front garment image is required, the rest are optional.
53
  Default number of images is 4.
54
  """
55
  request_id = str(uuid.uuid4())
@@ -212,6 +212,7 @@ async def check_status(request_id: str):
212
  @app.get("/status/{request_id}")
213
  async def check_status(request_id: str):
214
  output_images = []
 
215
  status = r.get(request_id)
216
  images = ast.literal_eval( r.get(request_id + "_content"))
217
  for image in images['images']:
@@ -223,7 +224,6 @@ async def check_status(request_id: str):
223
  output_images.append(image['url'])
224
  except Exception as e:
225
  print("no variations")
226
- variations = []
227
  prompt = r.get(request_id + "_prompt")
228
  result = r.get(request_id + "_result")
229
  error = r.get(request_id + "_error")
 
48
  garment_image_3: UploadFile = File(None),
49
  garment_image_4: UploadFile = File(None),):
50
  """
51
+ Pass at least the person_image and one garment_image. Pose is random (from the poses.py file). if not enough garments are provided, the system will generate the rest of the outfit.
52
+ Variations are created after base images are generated. sometimes they get a bit soft, but they produce interesting results.
53
  Default number of images is 4.
54
  """
55
  request_id = str(uuid.uuid4())
 
212
  @app.get("/status/{request_id}")
213
  async def check_status(request_id: str):
214
  output_images = []
215
+ variations = []
216
  status = r.get(request_id)
217
  images = ast.literal_eval( r.get(request_id + "_content"))
218
  for image in images['images']:
 
224
  output_images.append(image['url'])
225
  except Exception as e:
226
  print("no variations")
 
227
  prompt = r.get(request_id + "_prompt")
228
  result = r.get(request_id + "_result")
229
  error = r.get(request_id + "_error")