Spaces:
Sleeping
Sleeping
Update text_generator.py
Browse files- text_generator.py +1 -1
text_generator.py
CHANGED
@@ -14,7 +14,7 @@ class TextGenerationTool(Tool):
|
|
14 |
outputs = ["text"]
|
15 |
|
16 |
def __call__(self, prompt: str):
|
17 |
-
API_URL = "https://api-inference.huggingface.co/models/
|
18 |
headers = {"Authorization": "Bearer " + os.environ['hf']}
|
19 |
|
20 |
# Define the payload for the request
|
|
|
14 |
outputs = ["text"]
|
15 |
|
16 |
def __call__(self, prompt: str):
|
17 |
+
API_URL = "https://api-inference.huggingface.co/models/gpt2"
|
18 |
headers = {"Authorization": "Bearer " + os.environ['hf']}
|
19 |
|
20 |
# Define the payload for the request
|