代码初始化

picture/generalImg.py

@@ -1,7 +1,7 @@
 import base64
 from fastapi import APIRouter, HTTPException
 from fastapi import Body
-from huggingface_hub import InferenceClient
+from huggingface_hub import InferenceClient, TextToImageTargetSize
 from io import BytesIO
 
 routerImg = APIRouter()
@@ -13,8 +13,10 @@ async def generate_image(token: str = Body(...), prompt: str = Body(...)):
         # 创建InferenceClient
         client = InferenceClient("black-forest-labs/FLUX.1-dev", token=token)
 
+        targetSize = TextToImageTargetSize(height=512, width=512)
+
         # 使用text_to_image方法生成图片
-        image = client.text_to_image(prompt)
+        image = client.text_to_image(prompt=prompt, target_size=targetSize)
 
         # 将PIL.Image对象转换为byte数据
         buffered = BytesIO()