random2222 commited on
Commit
89fbd84
·
verified ·
1 Parent(s): 7c64e4c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -5
app.py CHANGED
@@ -29,8 +29,8 @@ def create_qa_system():
29
 
30
  # Text splitting
31
  text_splitter = CharacterTextSplitter(
32
- chunk_size=800,
33
- chunk_overlap=100,
34
  separator="\n\n"
35
  )
36
  texts = text_splitter.split_documents(documents)
@@ -49,8 +49,10 @@ def create_qa_system():
49
  "text2text-generation",
50
  model="google/flan-t5-base",
51
  tokenizer=tokenizer,
52
- max_length=300,
53
- temperature=0.3,
 
 
54
  device=-1
55
  )
56
 
@@ -60,7 +62,7 @@ def create_qa_system():
60
  # Create QA chain
61
  return RetrievalQA.from_llm(
62
  llm=llm,
63
- retriever=db.as_retriever(search_kwargs={"k": 2}),
64
  return_source_documents=True
65
  )
66
  except Exception as e:
 
29
 
30
  # Text splitting
31
  text_splitter = CharacterTextSplitter(
32
+ chunk_size=1100,
33
+ chunk_overlap=200,
34
  separator="\n\n"
35
  )
36
  texts = text_splitter.split_documents(documents)
 
49
  "text2text-generation",
50
  model="google/flan-t5-base",
51
  tokenizer=tokenizer,
52
+ max_length=600,
53
+ temperature=0.7,
54
+ do_sample=True,
55
+ top_k=50,
56
  device=-1
57
  )
58
 
 
62
  # Create QA chain
63
  return RetrievalQA.from_llm(
64
  llm=llm,
65
+ retriever=db.as_retriever(search_kwargs={"k": 3}),
66
  return_source_documents=True
67
  )
68
  except Exception as e: