Update text_generator.py

text_generator.py

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