Spaces:
Sleeping
Sleeping
chips commited on
Commit ·
f2e5577
1
Parent(s): faf3fab
more on variations
Browse files
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
|
| 52 |
-
|
| 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")
|