Spaces:
Sleeping
Sleeping
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 :)")
|