代码初始化

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": "system",
+            "role": "assistant",
             "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."
@@ -88,9 +88,10 @@ async def chat_completion(token: str = Body(...), messages: list = Body(...)):
         print("问题：", messages)
         messages.append({
             "role": "system",
-            "content": "You are a highly intelligent image generation text optimizer. Your role is to take the user's "
-                       "input prompt and enhance it, ensuring it is precise, vivid, and detailed, while maintaining "
-                       "its intended meaning. Always provide the enhanced version in English."
+            "content": "You are a highly intelligent image generation text optimizer. Your role is to enhance the "
+                       "user's input prompt, ensuring it is precise, vivid, and detailed, while maintaining its "
+                       "original intent. Always provide the enhanced version in English. Only reply with the "
+                       "optimized prompt."
         })
 
         # 使用 chat API 请求生成模型的回答  备选模型  google/gemma-2-2b-it