hamzafar commited on
Commit
2bd528d
·
verified ·
1 Parent(s): d373a5f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -25,13 +25,13 @@ class DocumentProcessor:
25
  self.persist_directory = './docs/chroma/'
26
  # self.vectordb = None
27
 
28
- # self.llm = HuggingFaceHub(repo_id="tiiuae/falcon-7b-instruct",
29
- # model_kwargs={"max_length": 300, "max_new_tokens": 300})
30
  # self.llm = HuggingFaceHub(repo_id="tiiuae/falcon-7b-instruct",
31
  # model_kwargs={"max_length": 1000, "max_new_tokens": 1000})
32
 
33
- self.llm = HuggingFaceHub(repo_id="tiiuae/falcon-40b", # tiiuae/falcon-40b
34
- model_kwargs={"max_length": 1000, "max_new_tokens": 1000})
35
 
36
  def load_documents(self, file_paths):
37
  # self.document_paths = [r"E:\FreeLance\download\tmp_doc.pdf"]
 
25
  self.persist_directory = './docs/chroma/'
26
  # self.vectordb = None
27
 
28
+ self.llm = HuggingFaceHub(repo_id="tiiuae/falcon-7b-instruct",
29
+ model_kwargs={"max_length": 300, "max_new_tokens": 300})
30
  # self.llm = HuggingFaceHub(repo_id="tiiuae/falcon-7b-instruct",
31
  # model_kwargs={"max_length": 1000, "max_new_tokens": 1000})
32
 
33
+ # self.llm = HuggingFaceHub(repo_id="tiiuae/falcon-40b", # tiiuae/falcon-40b
34
+ # model_kwargs={"max_length": 1000, "max_new_tokens": 1000})
35
 
36
  def load_documents(self, file_paths):
37
  # self.document_paths = [r"E:\FreeLance\download\tmp_doc.pdf"]