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={
|