Update app.py

app.py

@@ -48,8 +48,8 @@ def qwen_api(user_message, top_p=0.9,temperature=0.7, system_message='', max_tok
 
     response = ""
     for message in  client.chat.completions.create(
-        # model="meta-llama/Meta-Llama-3-8B-Instruct",
-        model="Qwen/Qwen1.5-4B-Chat",
+        model="meta-llama/Meta-Llama-3-8B-Instruct",
+        # model="Qwen/Qwen1.5-4B-Chat",
         max_tokens=max_tokens,
         stream=True,
         temperature=temperature,
@@ -70,8 +70,9 @@ embedding = load_embedding_mode()
 db = Chroma(persist_directory='./VecterStore2_512_txt/VecterStore2_512_txt', embedding_function=embedding)
 prompt_template = """
     {context}
-    The above content is a form of biological background knowledge. Please answer the questions according to the above content. Please be sure to answer the questions according to the background knowledge and attach the doi number of the information source when answering.
+    The above content is a form of biological background knowledge. Please answer the questions according to the above content.
     Question: {question}
+    Please be sure to answer the questions according to the background knowledge and attach the doi number of the information source when answering.
     Answer in English:"""
 PROMPT = PromptTemplate(
     template=prompt_template, input_variables=["context", "question"]
@@ -134,8 +135,8 @@ def respond(
     response = ""
     
     for message in  client.chat.completions.create(
-        # model="meta-llama/Meta-Llama-3-8B-Instruct",
-        model="Qwen/Qwen1.5-4B-Chat",
+        model="meta-llama/Meta-Llama-3-8B-Instruct",
+        # model="Qwen/Qwen1.5-4B-Chat",
         max_tokens=max_tokens,
         stream=True,
         temperature=temperature,