most-downloaded-model

Sleeping

Chris4K commited on Nov 21, 2023

Commit

977090c

•

1 Parent(s): 299669a

Update text_generator.py

Files changed (1) hide show

text_generator.py CHANGED Viewed

@@ -28,7 +28,7 @@ class TextGenerationTool(Tool):
         #def query(payload):
         generated_text = requests.post(API_URL, headers=headers, json=payload).json()
         print(generated_text)
-        return generated_text["generated_text"]

         #def query(payload):
         generated_text = requests.post(API_URL, headers=headers, json=payload).json()
         print(generated_text)
+        return generated_text["text"]