Spaces:
Sleeping
Sleeping
Commit ·
341fff4
1
Parent(s): d32f801
UPDATE:pipeline
Browse files- config.ini +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 =
|
| 4 |
normalize_embeddings = true
|
| 5 |
|
| 6 |
[VECTORSTORE]
|
|
@@ -22,4 +22,4 @@ fetchK = 10
|
|
| 22 |
timeout = 30
|
| 23 |
|
| 24 |
[EASYOCR]
|
| 25 |
-
gpu =
|
|
|
|
| 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
|