Workin on it

routers/training.py

@@ -56,8 +56,8 @@ async def performAction(actionBody: ActionBody):
     init_image = init_image.resize((actionBody.resizeW, actionBody.resizeH))
 
     response = requests.get(actionBody.maskUrl)
-    init_image = Image.open(BytesIO(response.content)).convert("RGB")
-    init_image = init_image.resize((actionBody.resizeW, actionBody.resizeH))
+    mask_image = Image.open(BytesIO(response.content)).convert("RGB")
+    mask_image = mask_image.resize((actionBody.resizeW, actionBody.resizeH))
 
     
     images = pipe(prompt=actionBody.prompt, image=init_image,  mask_image=mask_image, strength=actionBody.strength, guidance_scale=actionBody.guidance_scale).images