Spaces:

Chris4K
/

agent-reference-implemenation

Sleeping

Chris4K commited on Jan 20, 2024

Commit

2ea8b67

verified ·

1 Parent(s): 47cf4e1

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -121,7 +121,8 @@ def handle_submission():
     print(selected_tools)
     # Initialize the agent
     agent = CustomHfAgent(
-        url_endpoint="https://api-inference.huggingface.co/models/bigcode/starcoder",
         token=os.environ['HF_token'],
         additional_tools=selected_tools,
         input_params={"max_new_tokens": 192},

     print(selected_tools)
     # Initialize the agent
     agent = CustomHfAgent(
+        #url_endpoint="https://api-inference.huggingface.co/models/bigcode/starcoder", # mistralai/Mixtral-8x7B-v0.1
+        url_endpoint="https://api-inference.huggingface.co/models/mistralai/Mixtral-8x7B-v0.1", #
         token=os.environ['HF_token'],
         additional_tools=selected_tools,
         input_params={"max_new_tokens": 192},