Spaces:
Running
Running
Update main.py
Browse files
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={
|