Update app.py
Browse files
app.py
CHANGED
@@ -11,6 +11,7 @@ HF_SPACES_API_KEY = st.secrets["HF_token"]
|
|
11 |
API_URL = "https://api-inference.huggingface.co/models/microsoft/prophetnet-large-uncased-squad-qg"
|
12 |
headers = {"Authorization": HF_SPACES_API_KEY}
|
13 |
|
|
|
14 |
def query(payload):
|
15 |
response = requests.post(API_URL, headers=headers, json=payload)
|
16 |
return response.json()
|
@@ -21,6 +22,7 @@ def query(payload):
|
|
21 |
API_URL_evidence = "https://api-inference.huggingface.co/models/google/flan-t5-xxl"
|
22 |
headers_evidence = {"Authorization": HF_SPACES_API_KEY}
|
23 |
|
|
|
24 |
def query_evidence(payload):
|
25 |
response = requests.post(API_URL_evidence, headers=headers_evidence, json=payload)
|
26 |
return response.json()
|
|
|
11 |
API_URL = "https://api-inference.huggingface.co/models/microsoft/prophetnet-large-uncased-squad-qg"
|
12 |
headers = {"Authorization": HF_SPACES_API_KEY}
|
13 |
|
14 |
+
@st.cache_data
|
15 |
def query(payload):
|
16 |
response = requests.post(API_URL, headers=headers, json=payload)
|
17 |
return response.json()
|
|
|
22 |
API_URL_evidence = "https://api-inference.huggingface.co/models/google/flan-t5-xxl"
|
23 |
headers_evidence = {"Authorization": HF_SPACES_API_KEY}
|
24 |
|
25 |
+
@st.cache_data
|
26 |
def query_evidence(payload):
|
27 |
response = requests.post(API_URL_evidence, headers=headers_evidence, json=payload)
|
28 |
return response.json()
|