Spaces:
Sleeping
Sleeping
Commit
·
96961ef
1
Parent(s):
0779bba
update: different clothing
Browse files
app.py
CHANGED
@@ -40,7 +40,7 @@ class ClothingTryOnEntity(BaseModel):
|
|
40 |
|
41 |
@app.post("/clothingTryOn")
|
42 |
async def clothing_try_on(image: UploadFile = File(...),
|
43 |
-
mask: UploadFile = File(...)):
|
44 |
image_bytes = await image.read()
|
45 |
mask_bytes = await mask.read()
|
46 |
image, mask = Image.open(BytesIO(image_bytes)).convert("RGB"), Image.open(
|
@@ -74,7 +74,7 @@ async def clothing_try_on(image: UploadFile = File(...),
|
|
74 |
input = {
|
75 |
"mask": mask_data_uri,
|
76 |
"image": image_data_uri,
|
77 |
-
"prompt": "
|
78 |
"negative_prompt": "necklaces, jewellery, jewelry, necklace, neckpiece, garland, chain, neck wear, jewelled neck, jeweled neck, necklace on neck, jewellery on neck, accessories, watermark, text, changed background, wider body, narrower body, bad proportions, extra limbs, mutated hands, changed sizes, altered proportions, unnatural body proportions, blury, ugly",
|
79 |
"num_inference_steps": 25
|
80 |
}
|
@@ -303,4 +303,4 @@ async def necklace_try_on_id(necklace_try_on_id: NecklaceTryOnIDEntity = Depends
|
|
303 |
return JSONResponse(content=response)
|
304 |
|
305 |
else:
|
306 |
-
return JSONResponse(content=response)
|
|
|
40 |
|
41 |
@app.post("/clothingTryOn")
|
42 |
async def clothing_try_on(image: UploadFile = File(...),
|
43 |
+
mask: UploadFile = File(...), clothing_type: str = Form(...)):
|
44 |
image_bytes = await image.read()
|
45 |
mask_bytes = await mask.read()
|
46 |
image, mask = Image.open(BytesIO(image_bytes)).convert("RGB"), Image.open(
|
|
|
74 |
input = {
|
75 |
"mask": mask_data_uri,
|
76 |
"image": image_data_uri,
|
77 |
+
"prompt": f"Dull {clothing_type}, non-reflective clothing, properly worn, natural setting, elegant, natural look, neckline without jewellery, simple, perfect eyes, perfect face, perfect body, high quality, realistic, photorealistic, high resolution,traditional full sleeve blouse",
|
78 |
"negative_prompt": "necklaces, jewellery, jewelry, necklace, neckpiece, garland, chain, neck wear, jewelled neck, jeweled neck, necklace on neck, jewellery on neck, accessories, watermark, text, changed background, wider body, narrower body, bad proportions, extra limbs, mutated hands, changed sizes, altered proportions, unnatural body proportions, blury, ugly",
|
79 |
"num_inference_steps": 25
|
80 |
}
|
|
|
303 |
return JSONResponse(content=response)
|
304 |
|
305 |
else:
|
306 |
+
return JSONResponse(content=response)
|