techconsptrs commited on
Commit
341fff4
·
1 Parent(s): d32f801

UPDATE:pipeline

Browse files
Files changed (2) hide show
  1. config.ini +2 -2
  2. src/pipelines/completePipeline.py +0 -1
config.ini CHANGED
@@ -1,6 +1,6 @@
1
  [EMBEDDINGS]
2
  embeddingModel = BAAI/bge-large-en-v1.5
3
- device = cuda
4
  normalize_embeddings = true
5
 
6
  [VECTORSTORE]
@@ -22,4 +22,4 @@ fetchK = 10
22
  timeout = 30
23
 
24
  [EASYOCR]
25
- gpu = true
 
1
  [EMBEDDINGS]
2
  embeddingModel = BAAI/bge-large-en-v1.5
3
+ device = cpu
4
  normalize_embeddings = true
5
 
6
  [VECTORSTORE]
 
22
  timeout = 30
23
 
24
  [EASYOCR]
25
+ gpu = false
src/pipelines/completePipeline.py CHANGED
@@ -34,6 +34,5 @@ class Pipeline:
34
 
35
  def youtubeLinks(self, urls: list[str]):
36
  extractedText = self.youtubeLoader.getTranscripts(urls = urls)
37
- print(extractedText)
38
  chain = self.ragChain.returnChain(text = extractedText)
39
  return chain
 
34
 
35
  def youtubeLinks(self, urls: list[str]):
36
  extractedText = self.youtubeLoader.getTranscripts(urls = urls)
 
37
  chain = self.ragChain.returnChain(text = extractedText)
38
  return chain