Spaces:
Runtime error
Runtime error
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",
|