Final_Assignment_Template

Running

IamRulo commited on 8 days ago

Commit

f9e1e10

verified ·

1 Parent(s): 2468140

Update agent.py

Files changed (1) hide show

agent.py CHANGED Viewed

@@ -180,7 +180,7 @@ def build_graph(provider: str = "huggingface"):
         # TODO: Add huggingface endpoint
         llm = ChatHuggingFace(
             llm=HuggingFaceEndpoint(
-                url="https://api-inference.huggingface.co/models/Meta-DeepLearning/llama-2-7b-chat-hf",
                 temperature=0.2,
             ),
         )

         # TODO: Add huggingface endpoint
         llm = ChatHuggingFace(
             llm=HuggingFaceEndpoint(
+                endpoint_url="https://api-inference.huggingface.co/models/Meta-DeepLearning/llama-2-7b-chat-hf",
                 temperature=0.2,
             ),
         )