hae

model/custom_agent.py

@@ -75,7 +75,7 @@ class CustomHfAgent(Agent):
         Returns:
         - str: The generated response.
         """
-        headers = {"Authorization": self.token}
+        headers = {"Authorization": "Bearer " +self.token}
         max_new_tokens = self.input_params.get("max_new_tokens", 192)
         parameters = {"max_new_tokens": max_new_tokens, "return_full_text": False, "stop": stop, "padding": True, "truncation": True}
         inputs = {