Update app.py
Browse files
app.py
CHANGED
@@ -106,8 +106,9 @@ def handle_submission():
|
|
106 |
print(selected_tools)
|
107 |
# Initialize the agent
|
108 |
agent = CustomHfAgent(
|
109 |
-
#url_endpoint="https://api-inference.huggingface.co/models/bigcode/starcoder",
|
110 |
-
|
|
|
111 |
token=os.environ['HF_token'],
|
112 |
additional_tools=selected_tools,
|
113 |
input_params={"max_new_tokens": 192},
|
|
|
106 |
print(selected_tools)
|
107 |
# Initialize the agent
|
108 |
agent = CustomHfAgent(
|
109 |
+
#url_endpoint="https://api-inference.huggingface.co/models/bigcode/starcoder",
|
110 |
+
# deepseek-ai/deepseek-coder-6.7b-instruct # mistralai/Mixtral-8x7B-v0.1 # bigscience/bloom
|
111 |
+
url_endpoint="https://api-inference.huggingface.co/models/deepseek-ai/deepseek-coder-6.7b-instruct", #
|
112 |
token=os.environ['HF_token'],
|
113 |
additional_tools=selected_tools,
|
114 |
input_params={"max_new_tokens": 192},
|