Spaces:
Sleeping
Sleeping
Update text_generator.py
Browse files- 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['
|
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 |
#}
|