Chris4K commited on
Commit
bbb822d
·
verified ·
1 Parent(s): e443db2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -108,7 +108,7 @@ def handle_submission():
108
  agent = CustomHfAgent(
109
  #url_endpoint="https://api-inference.huggingface.co/models/bigcode/starcoder",
110
  # codellama/CodeLlama-7b-Instruct-hf # WizardLM/WizardCoder-Python-34B-V1.0 # deepseek-ai/deepseek-coder-6.7b-instruct # mistralai/Mixtral-8x7B-v0.1 # bigscience/bloom
111
- url_endpoint="https://api-inference.huggingface.co/models/codellama/CodeLlama-7b-Instruct-hf", #
112
  token=os.environ['HF_token'],
113
  additional_tools=selected_tools,
114
  input_params={"max_new_tokens": 192},
 
108
  agent = CustomHfAgent(
109
  #url_endpoint="https://api-inference.huggingface.co/models/bigcode/starcoder",
110
  # codellama/CodeLlama-7b-Instruct-hf # WizardLM/WizardCoder-Python-34B-V1.0 # deepseek-ai/deepseek-coder-6.7b-instruct # mistralai/Mixtral-8x7B-v0.1 # bigscience/bloom
111
+ url_endpoint="https://api-inference.huggingface.co/models/bigcode/starcoder", #
112
  token=os.environ['HF_token'],
113
  additional_tools=selected_tools,
114
  input_params={"max_new_tokens": 192},