Chris4K commited on
Commit
280c002
·
verified ·
1 Parent(s): ef549d1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
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", # mistralai/Mixtral-8x7B-v0.1 # bigscience/bloom
110
- url_endpoint="https://api-inference.huggingface.co/models/bigcode/starcoder", #
 
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},