feat/todo

app.py

@@ -35,8 +35,8 @@ model = AutoModelForCausalLM.from_pretrained(
 )
 
 SYS = (
-    "You are a legal AI assistant specialized in GDPR/EDPB. "
-    "If you cannot find an answer in the context, it's okay to speculate. But if so, make it clear. "
+    "You are a legal AI assistant specialized in GDPR/EDPB."
+    "If you cannot find an answer in the context, it's okay to speculate. But if so, make it clear."
     "Answer this Question:"
 )
 
@@ -101,11 +101,13 @@ def qa_fn(question, top_k, temperature, max_tokens):
         output = output.split("Output:", 1)[1].strip()
     return "\n# Generated Answer\n", output,"\n# Used Documents\n",  file_links, "\n# Used Context\n", markdown_chunks
 
+# TODO first model to make prompt more efficient
+
 heading_answer  = gr.Markdown(label="Answer Heading")
 outputs_answer = gr.Textbox(label="Answer")
-outputs_link = gr.Markdown(label="Source Link")
 heading_links  = gr.Markdown(label="Links Heading")
 heading_chunks = gr.Markdown(label="Chunks Heading")
+outputs_link = gr.Markdown(label="Source Link")
 outputs_chunks = gr.Markdown(label="Used Chunks")
 
 demo = gr.Interface(