Chris4K commited on
Commit
953ea93
·
1 Parent(s): 2c23988
Files changed (1) hide show
  1. model/custom_agent.py +1 -1
model/custom_agent.py CHANGED
@@ -75,7 +75,7 @@ class CustomHfAgent(Agent):
75
  Returns:
76
  - str: The generated response.
77
  """
78
- headers = {"Authorization": self.token}
79
  max_new_tokens = self.input_params.get("max_new_tokens", 192)
80
  parameters = {"max_new_tokens": max_new_tokens, "return_full_text": False, "stop": stop, "padding": True, "truncation": True}
81
  inputs = {
 
75
  Returns:
76
  - str: The generated response.
77
  """
78
+ headers = {"Authorization": "Bearer " +self.token}
79
  max_new_tokens = self.input_params.get("max_new_tokens", 192)
80
  parameters = {"max_new_tokens": max_new_tokens, "return_full_text": False, "stop": stop, "padding": True, "truncation": True}
81
  inputs = {