Update main.py

main.py

@@ -159,7 +159,7 @@ async def api_generate(request: Request, inputs: ApiInput):
         # 'fine_weave_image': to_base64(final_results['fine_weave_image']),
         # 'bold_statement_image': to_base64(final_results['bold_statement_image']),
         # 'creative_variation_image': to_base64(final_results['creative_variation_image']),
-        # 'mask_image': to_base64(mask)
+        'mask_image': to_base64(mask)
     }
     
     return response_data