Spaces:
Paused
Paused
auth update
Browse files
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 |
|