only show the generated string

app.py

@@ -114,7 +114,7 @@ real_docstring = r"""
 def docgen_func(function_code):
     req_data = {"inputs": function_code}
     output = query(req_data)
-    return output['generated_text']
+    return output[0]['generated_text']
 
 def pygen_func(nl_code_intent):
     pass # TODO: generate code PL from intent NL + search in corpus