miwojc commited on
Commit
8a56d89
·
1 Parent(s): 5b2d028

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -1
app.py CHANGED
@@ -9,11 +9,15 @@ PROMPT_LIST = {
9
  "Cześć, mam na imię...": ["Cześć, mam na imię "],
10
  "Największym polskim poetą był...": ["Największym polskim poetą był "],
11
  }
 
 
 
 
12
  def query(payload, model_url):
13
  data = json.dumps(payload)
14
  print("model url:", model_url)
15
  response = requests.request(
16
- "POST", model_url, headers={}, data=data
17
  )
18
  return json.loads(response.content.decode("utf-8"))
19
  def process(
 
9
  "Cześć, mam na imię...": ["Cześć, mam na imię "],
10
  "Największym polskim poetą był...": ["Największym polskim poetą był "],
11
  }
12
+
13
+ API_TOKEN = st.secrets['api_token']
14
+ headers = {"Authorization": f"Bearer {API_TOKEN}"}
15
+
16
  def query(payload, model_url):
17
  data = json.dumps(payload)
18
  print("model url:", model_url)
19
  response = requests.request(
20
+ "POST", model_url, headers=headers, data=data
21
  )
22
  return json.loads(response.content.decode("utf-8"))
23
  def process(