Update app.py
Browse files
app.py
CHANGED
|
@@ -20,7 +20,7 @@ def query_model(model_id, prompt, max_tokens=500, temperature=0.7):
|
|
| 20 |
"""
|
| 21 |
Query a model using the direct Inference API endpoint
|
| 22 |
"""
|
| 23 |
-
API_URL = f"https://
|
| 24 |
headers = {"Authorization": f"Bearer {HF_TOKEN}"}
|
| 25 |
|
| 26 |
payload = {
|
|
|
|
| 20 |
"""
|
| 21 |
Query a model using the direct Inference API endpoint
|
| 22 |
"""
|
| 23 |
+
API_URL = f"https://router.huggingface.co/models/{model_id}"
|
| 24 |
headers = {"Authorization": f"Bearer {HF_TOKEN}"}
|
| 25 |
|
| 26 |
payload = {
|