Update app.py

app.py

@@ -113,17 +113,17 @@ if INTdesc_input is not None and createSBScodes_clicked == True:
     question = "Which, if any, of the below Saudi Billing System descriptions corresponds best to " + INTdesc_input +"? " 
     shortlist = [SBScorpus[result[0]["corpus_id"]], SBScorpus[result[1]["corpus_id"]], SBScorpus[result[2]["corpus_id"]], SBScorpus[result[3]["corpus_id"]], SBScorpus[result[4]["corpus_id"]]] 
     prompt = [question + " " + shortlist[0] + " " + shortlist[1] + " " + shortlist[2] + " " + shortlist[3] + " " + shortlist[4]]
-    st.write(prompt)
+    #st.write(prompt)
     
     messages = [
-    {"role": "system", "content": "You are a pirate chatbot who always responds in pirate speak!"},
-    {"role": "user", "content": "Who are you?"},
+    {"role": "system", "content": "You are a knowledgable AI assistant who always answers truthfully and precisely!"},
+    {"role": "user", "content": prompt},
     ]
     outputs = pipe(
         messages,
         max_new_tokens=256,
     )
-    st.write(outputs[0]["generated_text"][-1])
+    st.write(outputs[0]["generated_text"][-1]["content"])
     
     bs, b1, b2, b3, bLast = st.columns([0.75, 1.5, 1.5, 1.5, 0.75])
     with b1: