codelion commited on
Commit
5516e24
·
verified ·
1 Parent(s): 8bbca49

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +2 -1
main.py CHANGED
@@ -62,7 +62,8 @@ def response_generator(query: str) -> str:
62
 
63
  # prepare HF text-generation LLM
64
  hf = HuggingFaceEndpoint(
65
- endpoint_url=f"https://api-inference.huggingface.co/models/{model}",
 
66
  task="text-generation",
67
  huggingfacehub_api_token=hf_api_key,
68
  model_kwargs={
 
62
 
63
  # prepare HF text-generation LLM
64
  hf = HuggingFaceEndpoint(
65
+ # endpoint_url=f"https://api-inference.huggingface.co/models/{model}",
66
+ endpoint_url=f"https://router.huggingface.co/hf-inference/models/{model}",
67
  task="text-generation",
68
  huggingfacehub_api_token=hf_api_key,
69
  model_kwargs={