Update app.py

app.py

@@ -20,7 +20,6 @@ Paying tribute to the ancient master Zhang Zhongjing, this model integrates clas
 Please note!!! This model should not be used for any medical purposes or scenarios potentially involving medical or health advice. It is currently still in the research and testing stage. We sincerely request your valuable feedback. Thank you.
 """
 
-
 LICENSE = """
 <p/>
 
@@ -38,11 +37,14 @@ if torch.cuda.is_available():
     tokenizer = AutoTokenizer.from_pretrained(model_id)
     tokenizer.use_default_system_prompt = False
 
+# 定义 system_prompt 变量
+system_prompt = "You are a helpful TCM assistant named 仲景中医大语言模型, created by 医哲未来. You can switch between Chinese and English based on user preference."
+
 @spaces.GPU
 def generate(
     message: str,
     chat_history: list[tuple[str, str]],
-    system_prompt: str = "You are a helpful TCM assistant named 仲景中医大语言模型, created by 医哲未来. You can switch between Chinese and English based on user preference.",
+    system_prompt: str = system_prompt,
     max_new_tokens: int = 1024,
     temperature: float = 0.6,
     top_p: float = 0.95,
@@ -83,7 +85,11 @@ def generate(
 chat_interface = gr.ChatInterface(
     fn=generate,
     additional_inputs=[
-        gr.Textbox(label="System prompt", lines=6, value="You are a helpful TCM assistant named 仲景中医大语言模型, created by 医哲未来. You can switch between Chinese and English based on user preference."),
+        gr.Textbox(
+            label="System prompt",
+            lines=6,
+            value=system_prompt,
+        ),
         gr.Slider(
             label="Max new tokens",
             minimum=1,