merve HF staff commited on
Commit
39fa1f9
·
1 Parent(s): 050bcdd

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -8,9 +8,9 @@ from streamlit_chat import message
8
  def query(payload):
9
  api_token = os.getenv("api_token")
10
  model_id = "deepset/roberta-base-squad2"
11
- headers = {"Authorization": f"Bearer {api_token}"}
12
- API_URL = f"https://api-inference.huggingface.co/models/{model_id}"
13
- response = requests.post(API_URL, headers=headers, json=payload)
14
  return response.json()
15
 
16
  message("Let's find out the best task for your use case! Tell me about your use case :)")
 
8
  def query(payload):
9
  api_token = os.getenv("api_token")
10
  model_id = "deepset/roberta-base-squad2"
11
+ headers = {"Authorization": f"Bearer {api_token}"}
12
+ API_URL = f"https://api-inference.huggingface.co/models/{model_id}"
13
+ response = requests.post(API_URL, headers=headers, json=payload)
14
  return response.json()
15
 
16
  message("Let's find out the best task for your use case! Tell me about your use case :)")