Spaces:

Chris4K
/

text-generation-tool

Sleeping

Chris4K commited on Nov 21, 2023

Commit

3a5fb85

1 Parent(s): f275817

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/mistralai/Mistral-7B-v0.1"
         headers = {"Authorization": "Bearer " +  os.environ['hf']}
         payload = {

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