代码初始化

app.py

@@ -1,8 +1,10 @@
 from fastapi import FastAPI
 from inference.inference import router
+from picture.generalImg import routerImg
 
 app = FastAPI()
 app.include_router(router)
+app.include_router(routerImg)
 
 
 @app.get("/")

picture/generalImg.py

@@ -4,10 +4,10 @@ from fastapi import Body
 from huggingface_hub import InferenceClient
 from io import BytesIO
 
-router = APIRouter()
+routerImg = APIRouter()
 
 
-@router.post("/generate-image/")
+@routerImg.post("/generate-image/")
 async def generate_image(token: str = Body(...), prompt: str = Body(...)):
     try:
         # 创建InferenceClient