Workin on it

routers/training.py

@@ -60,7 +60,7 @@ async def performAction(actionBody: ActionBody):
     init_image = init_image.resize((actionBody.resizeW, actionBody.resizeH))
 
     
-    images = pipe(prompt=actionBody.prompt, image=init_image, strength=actionBody.strength, guidance_scale=actionBody.guidance_scale).images
+    images = pipe(prompt=actionBody.prompt, image=init_image,  mask_image=mask_image, strength=actionBody.strength, guidance_scale=actionBody.guidance_scale).images
     print(images)
     buffered = BytesIO()
     images[0].save(buffered, format="JPEG")