Update stools_client.py

stools_client.py

@@ -15,4 +15,14 @@ def _get_stools_client():
     global _stools_client
     if _stools_client is None:
         _stools_client = Client("VeuReu/stools")
-    return _stools_client
+    return _stools_client
+
+def get_from_prompt(prompt):
+    client = _get_stools_client()
+    
+    result = client.predict(
+		prompt=prompt,
+		api_name="/generate_out_from_prompt"
+    )
+
+    return result