Update text_generator.py

text_generator.py

@@ -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"]
+        return generated_text["text"]