text-generation-tool-duplicate

Sleeping

Chris4K commited on Nov 18, 2023

Commit

efe65ac

1 Parent(s): 126246c

Update text_generator.py

Files changed (1) hide show

text_generator.py CHANGED Viewed

@@ -14,7 +14,7 @@ class TextGenerationTool(Tool):
     outputs = ["text"]
     def __call__(self, prompt: str):
-        API_URL = "https://api-inference.huggingface.co/models/lukasdrg/clinical_longformer_same_tokens_220k"
         headers = {"Authorization": "Bearer " + os.environ['hf']}
         # Define the payload for the request

     outputs = ["text"]
     def __call__(self, prompt: str):
+        API_URL = "https://api-inference.huggingface.co/models/gpt2"
         headers = {"Authorization": "Bearer " + os.environ['hf']}
         # Define the payload for the request