Spaces:
Running
Running
Update app.py
Browse files
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 |
-
|
12 |
-
|
13 |
-
|
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 :)")
|