代码初始化

inference/inference.py

@@ -60,7 +60,7 @@ async def chat_completion(token: str = Body(...), messages: list = Body(...)):
         client = InferenceClient(api_key=token)
         print("问题：", messages)
         messages.append({
-            "role": "assistant",
+            "role": "system",
             "content": "You are a multilingual chatbot capable of understanding questions in various languages and "
                        "providing accurate responses in the appropriate language. However, all questions must be "
                        "answered in Chinese."
@@ -68,7 +68,7 @@ async def chat_completion(token: str = Body(...), messages: list = Body(...)):
 
         # 使用 chat API 请求生成模型的回答  备选模型  google/gemma-2-2b-it
         completion = client.chat.completions.create(
-            model="mistralai/Mistral-Nemo-Instruct-2407",
+            model="google/gemma-2-2b-it",
             messages=messages,
             max_tokens=500
         )

picture/generalImg.py

@@ -13,8 +13,7 @@ async def generate_image(token: str = Body(...), prompt: str = Body(...)):
         # 创建InferenceClient
         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")
+        image = client.text_to_image(prompt=prompt, height=256, width=256)
 
         # 将PIL.Image对象转换为byte数据
         buffered = BytesIO()