Update text_generator.py

text_generator.py

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