new code
Browse files- src/pdfchatbot.py +1 -0
src/pdfchatbot.py
CHANGED
|
@@ -54,6 +54,7 @@ class PDFChatBot:
|
|
| 54 |
self.vectordb = Chroma.from_documents(docs, self.embeddings)
|
| 55 |
print("Vector store created")
|
| 56 |
|
|
|
|
| 57 |
def create_organic_pipeline(self):
|
| 58 |
self.pipeline = pipeline(
|
| 59 |
"text-generation",
|
|
|
|
| 54 |
self.vectordb = Chroma.from_documents(docs, self.embeddings)
|
| 55 |
print("Vector store created")
|
| 56 |
|
| 57 |
+
@spaces.GPU
|
| 58 |
def create_organic_pipeline(self):
|
| 59 |
self.pipeline = pipeline(
|
| 60 |
"text-generation",
|