diff --git a/app/api/api_agent_generate_image.py b/app/api/api_agent_generate_image.py index d8efbbb..9aeda19 100644 --- a/app/api/api_agent_generate_image.py +++ b/app/api/api_agent_generate_image.py @@ -17,4 +17,4 @@ def generate_image(prompt: str): except Exception as e: logger.warning(f"generate_image Run Exception @@@@@@:{e}") raise HTTPException(status_code=404, detail=str(e)) - return ResponseModel(data=data) + return data diff --git a/app/api/api_route.py b/app/api/api_route.py index 61bd43f..45f8567 100644 --- a/app/api/api_route.py +++ b/app/api/api_route.py @@ -30,6 +30,6 @@ router.include_router(api_image2sketch.router, tags=['api_image2sketch'], prefix router.include_router(api_brighten.router, tags=['api_brighten'], prefix="/api") router.include_router(api_query_image.router, tags=['api_query_image'], prefix="/api") router.include_router(api_brand_dna.router, tags=['api_brand_dna'], prefix="/api") -router.include_router(api_recommendation.router, tags=['api_recommendation'], prefix="/api") +# router.include_router(api_recommendation.router, tags=['api_recommendation'], prefix="/api") router.include_router(api_mannequins_edit.router, tags=['api_mannequins_edit'], prefix="/api") router.include_router(api_agent_generate_image.router, tags=['api_agent_generate_image'], prefix="/api")