Update text_generator.py

text_generator.py

@@ -15,7 +15,7 @@ class TextGenerationTool(Tool):
 
     def __call__(self, prompt: str):
         API_URL = "https://api-inference.huggingface.co/models/mistralai/Mistral-7B-v0.1"
-        headers = {"Authorization":  os.environ['hf']}
+        headers = {"Authorization": "Bearer " +  os.environ['hf']}
 
         payload = {
             "inputs": prompt  # Adjust this based on your model's input format