Towhidul commited on
Commit
d34e1aa
·
1 Parent(s): 31766d5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -4
app.py CHANGED
@@ -11,8 +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
- @st.cache_data
15
-
16
  def query(payload):
17
  response = requests.post(API_URL, headers=headers, json=payload)
18
  return response.json()
@@ -23,8 +22,7 @@ def query(payload):
23
  API_URL_evidence = "https://api-inference.huggingface.co/models/google/flan-t5-xxl"
24
  headers_evidence = {"Authorization": HF_SPACES_API_KEY}
25
 
26
- @st.cache_data
27
-
28
  def query_evidence(payload):
29
  response = requests.post(API_URL_evidence, headers=headers_evidence, json=payload)
30
  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(suppress_st_warning=True)
 
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(suppress_st_warning=True)
 
26
  def query_evidence(payload):
27
  response = requests.post(API_URL_evidence, headers=headers_evidence, json=payload)
28
  return response.json()