kambris commited on
Commit
22a1949
·
verified ·
1 Parent(s): 2b66fac

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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://api-inference.huggingface.co/models/{model_id}"
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 = {