代码初始化

picture/generalImg.py

@@ -11,7 +11,7 @@ routerImg = APIRouter()
 async def generate_image(token: str = Body(...), prompt: str = Body(...)):
     try:
         # 创建InferenceClient
-        client = InferenceClient("stabilityai/stable-diffusion", token=token)
+        client = InferenceClient("stabilityai/stable-diffusion-3-medium-diffusers", token=token)
         # 使用text_to_image方法生成图片
         image = client.text_to_image(prompt=prompt, height=256, width=256,
                                      negative_prompt="low quality, blurry, bad anatomy")