Spaces:
Sleeping
Sleeping
Update main.py
Browse files
main.py
CHANGED
|
@@ -108,7 +108,7 @@ import requests
|
|
| 108 |
|
| 109 |
HF_TOKEN = os.environ.get('HF_TOKEN')
|
| 110 |
headers = {"Authorization": f"Bearer {HF_TOKEN}"}
|
| 111 |
-
API_URL =
|
| 112 |
def query(payload):
|
| 113 |
response = requests.post(API_URL, headers=headers, json=payload)
|
| 114 |
return response.json()
|
|
|
|
| 108 |
|
| 109 |
HF_TOKEN = os.environ.get('HF_TOKEN')
|
| 110 |
headers = {"Authorization": f"Bearer {HF_TOKEN}"}
|
| 111 |
+
API_URL = os.environ.get('HF_API_URL')
|
| 112 |
def query(payload):
|
| 113 |
response = requests.post(API_URL, headers=headers, json=payload)
|
| 114 |
return response.json()
|