techconsptrs commited on
Commit
d14c53b
·
1 Parent(s): 341fff4
Files changed (2) hide show
  1. app.py +2 -1
  2. config.ini +2 -2
app.py CHANGED
@@ -179,6 +179,7 @@ def getLinksButtonFn(baseUrl: str):
179
  row3
180
  )
181
 
 
182
  def getWebsiteResponse(links: list[str], inputQuery: str):
183
  global chain
184
  if chain is None:
@@ -254,7 +255,7 @@ with gr.Blocks() as websiteCrawler:
254
  outputs = [checkboxes]
255
  )
256
 
257
-
258
  def getYoutubeResponse(links: str, inputQuery: str):
259
  global chain
260
  links = [link.strip() for link in links.split(",")]
 
179
  row3
180
  )
181
 
182
+ @spaces.GPU
183
  def getWebsiteResponse(links: list[str], inputQuery: str):
184
  global chain
185
  if chain is None:
 
255
  outputs = [checkboxes]
256
  )
257
 
258
+ @spaces.GPU
259
  def getYoutubeResponse(links: str, inputQuery: str):
260
  global chain
261
  links = [link.strip() for link in links.split(",")]
config.ini CHANGED
@@ -1,6 +1,6 @@
1
  [EMBEDDINGS]
2
  embeddingModel = BAAI/bge-large-en-v1.5
3
- device = cpu
4
  normalize_embeddings = true
5
 
6
  [VECTORSTORE]
@@ -22,4 +22,4 @@ fetchK = 10
22
  timeout = 30
23
 
24
  [EASYOCR]
25
- gpu = false
 
1
  [EMBEDDINGS]
2
  embeddingModel = BAAI/bge-large-en-v1.5
3
+ device = cuda
4
  normalize_embeddings = true
5
 
6
  [VECTORSTORE]
 
22
  timeout = 30
23
 
24
  [EASYOCR]
25
+ gpu = true