hanzla commited on
Commit
421861c
·
1 Parent(s): 0ecf283
Files changed (1) hide show
  1. 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",