Chris4K commited on
Commit
ef549d1
·
verified ·
1 Parent(s): 040497e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -107,10 +107,10 @@ def handle_submission():
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/bigscience/bloom", #
111
  token=os.environ['HF_token'],
112
  additional_tools=selected_tools,
113
- input_params={"max_new_tokens": 100},
114
  )
115
  #agent.tools = selected_tools
116
 
 
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},
114
  )
115
  #agent.tools = selected_tools
116