sanjeevbora commited on
Commit
46486f5
·
verified ·
1 Parent(s): 4164239

auth update

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -10,6 +10,7 @@ import torch
10
  import re
11
  import requests
12
  from urllib.parse import urlencode, urlparse, parse_qs
 
13
 
14
  # Step 1: Run the setup script
15
  script_path = './setup.sh' # Adjust the path if needed
@@ -121,6 +122,7 @@ def handle_redirect(url):
121
  return None
122
 
123
  # Function to retrieve answers using the RAG system
 
124
  def test_rag(query):
125
  books_retriever = books_db_client_retriever.run(query)
126
 
 
10
  import re
11
  import requests
12
  from urllib.parse import urlencode, urlparse, parse_qs
13
+ import spaces
14
 
15
  # Step 1: Run the setup script
16
  script_path = './setup.sh' # Adjust the path if needed
 
122
  return None
123
 
124
  # Function to retrieve answers using the RAG system
125
+ @spaces.GPU(duration=60)
126
  def test_rag(query):
127
  books_retriever = books_db_client_retriever.run(query)
128