Towhidul commited on
Commit
989d907
·
1 Parent(s): 0f98ea0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
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()