Update bookie.py
Browse files
bookie.py
CHANGED
@@ -13,7 +13,7 @@ import time
|
|
13 |
load_dotenv("bookie.env")
|
14 |
api_key=os.getenv("OPENAI_API_KEY")
|
15 |
api_base=os.getenv("OPENAI_API_BASE")
|
16 |
-
llm=ChatOpenAI(model_name="
|
17 |
em=joblib.load("bai.joblib")
|
18 |
mp=st.empty()
|
19 |
st.title("Welcome to Bookie ππ")
|
@@ -32,11 +32,11 @@ if upl and uploaded_file:
|
|
32 |
mp.text("loading doc")
|
33 |
loader = PyPDFLoader(tmp_path)
|
34 |
docs = loader.load()
|
35 |
-
st.write(len(docs))
|
36 |
mp.text("loading split")
|
37 |
tct=RecursiveCharacterTextSplitter.from_tiktoken_encoder(encoding_name="cl100k_base",chunk_size=512, chunk_overlap=16)
|
38 |
doc=tct.split_documents(docs)
|
39 |
-
st.write(len(doc))
|
40 |
mp.text("loading vector db")
|
41 |
vb= Chroma.from_documents(doc,em)
|
42 |
r1=vb.as_retriever(search_type="similarity",search_kwargs={"k":4})
|
@@ -56,12 +56,6 @@ if qb:
|
|
56 |
result=st.session_state.chain({"question":q},return_only_outputs=True)
|
57 |
st.header("Answer")
|
58 |
st.subheader(result["answer"])
|
59 |
-
sb=st.button("show sources")
|
60 |
-
if sb:
|
61 |
-
sources = result.get("sources", "")
|
62 |
-
st.subheader("Sources")
|
63 |
-
for line in sources.split("\n"):
|
64 |
-
st.write(line)
|
65 |
|
66 |
|
67 |
|
|
|
13 |
load_dotenv("bookie.env")
|
14 |
api_key=os.getenv("OPENAI_API_KEY")
|
15 |
api_base=os.getenv("OPENAI_API_BASE")
|
16 |
+
llm=ChatOpenAI(model_name="qwen/qwen3-coder:free",temperature=0.2)
|
17 |
em=joblib.load("bai.joblib")
|
18 |
mp=st.empty()
|
19 |
st.title("Welcome to Bookie ππ")
|
|
|
32 |
mp.text("loading doc")
|
33 |
loader = PyPDFLoader(tmp_path)
|
34 |
docs = loader.load()
|
35 |
+
st.write(len(docs))
|
36 |
mp.text("loading split")
|
37 |
tct=RecursiveCharacterTextSplitter.from_tiktoken_encoder(encoding_name="cl100k_base",chunk_size=512, chunk_overlap=16)
|
38 |
doc=tct.split_documents(docs)
|
39 |
+
st.write(len(doc))
|
40 |
mp.text("loading vector db")
|
41 |
vb= Chroma.from_documents(doc,em)
|
42 |
r1=vb.as_retriever(search_type="similarity",search_kwargs={"k":4})
|
|
|
56 |
result=st.session_state.chain({"question":q},return_only_outputs=True)
|
57 |
st.header("Answer")
|
58 |
st.subheader(result["answer"])
|
|
|
|
|
|
|
|
|
|
|
|
|
59 |
|
60 |
|
61 |
|