Update app.py

app.py

@@ -63,7 +63,7 @@ def load_model():
     prompt = ChatPromptTemplate.from_messages([
         HumanMessagePromptTemplate.from_template(
         f"""
-        Extract the appropriate 8-digit HS Code base on the product description and retrieved document by thoroughly analyzing its details and utilizing a reliable and up-to-date HS Code database for accurate results.
+        Extract the appropriate 6-digit HS Code base on the product description and retrieved document by thoroughly analyzing its details and utilizing a reliable and up-to-date HS Code database for accurate results.
         Only return the HS Code as a 6-digit number .
         Example: 123456
         Context: {{context}}
@@ -87,7 +87,7 @@ def process_input(sentence):
     docs, _ = st.session_state.retriever.retrieve(bm25s.tokenize(sentence), k=15)
     documents =[]
     for doc in docs[0]:
-        documents.append(Document(doc['text'])) 
+        documents.append(Document(doc)) 
     return documents
     
 if 'retriever' not in st.session_state: