Enoch1359 commited on
Commit
2903fbe
Β·
verified Β·
1 Parent(s): d4f56e6

Update bookie.py

Browse files
Files changed (1) hide show
  1. bookie.py +3 -9
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="google/gemma-3n-e2b-it:free",temperature=0.2)
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