Spaces:

CalebMaresca
/

AIE6-HF-endpoint-example

Sleeping

CalebMaresca commited on May 21, 2025

Commit

84e2c52

1 Parent(s): e4b83ec

oops

Files changed (1) hide show

app.py CHANGED Viewed

@@ -51,7 +51,7 @@ split_documents = text_splitter.split_documents(documents)
 ### 3. LOAD HUGGINGFACE EMBEDDINGS
 hf_embeddings = HuggingFaceEndpointEmbeddings(
-    model="https://vh1lkulhtegi6p9r.us-east-1.aws.endpoints.huggingface.cloud",
     task="feature-extraction",
     huggingfacehub_api_token=os.environ["HF_TOKEN"],
 )
@@ -137,7 +137,7 @@ rag_prompt = PromptTemplate.from_template(RAG_PROMPT_TEMPLATE)
 """
 ### 1. CREATE HUGGINGFACE ENDPOINT FOR LLM
 hf_llm = HuggingFaceEndpoint(
-    endpoint_url="https://bk4n5qv7rq0y3x5u.us-east-1.aws.endpoints.huggingface.cloud",
     task="text-generation",
     max_new_tokens=512,
     top_k=10,

 ### 3. LOAD HUGGINGFACE EMBEDDINGS
 hf_embeddings = HuggingFaceEndpointEmbeddings(
+    model=HF_EMBED_ENDPOINT,
     task="feature-extraction",
     huggingfacehub_api_token=os.environ["HF_TOKEN"],
 )
 """
 ### 1. CREATE HUGGINGFACE ENDPOINT FOR LLM
 hf_llm = HuggingFaceEndpoint(
+    endpoint_url=HF_LLM_ENDPOINT,
     task="text-generation",
     max_new_tokens=512,
     top_k=10,