Updated requirements and hf model

app.py

@@ -43,8 +43,11 @@ if file is not None:
             include=["documents", ]
         )
 
-        query_text = [" ".join([str(element) for element in inner_list])
-            for inner_list in query_results["documents"]][0]
-
         st.write("Database Query Matches")
-        st.markdown(query_text)
+        query_results
+
+        # query_text = [" ".join([str(element) for element in inner_list])
+        #     for inner_list in query_results["documents"]][0]
+
+        # st.write("Database Query Matches")
+        # st.markdown(query_text)

examples/llama3.py

@@ -1,7 +1,7 @@
 import transformers
 import torch
 
-model_id = "meta-llama/Meta-Llama-3-8B"
+model_id = "meta-llama/Meta-Llama-3.1-8B"
 
 pipeline = transformers.pipeline(
     "text-generation", model=model_id, model_kwargs={"torch_dtype": torch.bfloat16}, device_map="auto"

requirements.txt

@@ -1,3 +1,4 @@
 chromadb==0.5.5
 pymupdf==1.24.9
 streamlit==1.38.0
+transformers==4.44.2