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 |
|