Update main.py

main.py

@@ -156,10 +156,10 @@ async def api_generate(request: Request, inputs: ApiInput):
     # Prepare the response data with images converted to base64
     response_data = {
         'ultimate_image': to_base64(final_results['ultimate_image']),
-        '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)
+        # '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)
     }
     
     return response_data