ishworrsubedii commited on
Commit
96961ef
·
1 Parent(s): 0779bba

update: different clothing

Browse files
Files changed (1) hide show
  1. app.py +3 -3
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": "South Indian Saree, properly worn, natural setting, elegant, natural look, neckline without jewellery, simple",
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)