feat: imporve sysprompt

app.py

@@ -34,8 +34,7 @@ model = AutoModelForCausalLM.from_pretrained(
 
 SYS = (
 "You are a legal AI assistant specialized in GDPR/EDPB. "
-    "Answer the user's question directly based only on the provided context. Do not use  "
-    "If you cannot find an answer in the context, reply 'I do not know.'"
+    "If you cannot find an answer in the context, reply 'I do not know.' Answer this Question:"
 )
 
 def retrieve(q, k=3):
@@ -58,7 +57,7 @@ def retrieve(q, k=3):
 
 def make_prompt(q, docs):
     context = "\n\n".join(f"Title: {doc['title']}\nPages: {doc['pages']}" for doc in docs)
-    return f"Instruct:{SYS}\n\nContext:\n{context}\n\nQuestion:\n{q}\n\nOutput:"
+    return f"Instruct:{SYS} {q} based on the following documents:{context}\nOutput:"
 
 @spaces.GPU()
 def qa_fn(question, top_k, temperature, max_tokens):