Spaces:
Running
Running
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 |
#}
|