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