Chris4K commited on
Commit
16a9cb3
·
verified ·
1 Parent(s): 7e99192

Update text_generator.py

Browse files
Files changed (1) hide show
  1. text_generator.py +1 -1
text_generator.py CHANGED
@@ -18,7 +18,7 @@ class TextGenerationTool(Tool):
18
  def __call__(self, prompt: str):
19
  #API_URL = "https://api-inference.huggingface.co/models/openchat/openchat_3.5"
20
  #headers = {"Authorization": "Bearer " + os.environ['hf']}
21
- token=os.environ['hf']
22
  #payload = {
23
  # "inputs": prompt # Adjust this based on your model's input format
24
  #}
 
18
  def __call__(self, prompt: str):
19
  #API_URL = "https://api-inference.huggingface.co/models/openchat/openchat_3.5"
20
  #headers = {"Authorization": "Bearer " + os.environ['hf']}
21
+ token=os.environ['HF_token']
22
  #payload = {
23
  # "inputs": prompt # Adjust this based on your model's input format
24
  #}