update display string

app.py

@@ -185,8 +185,7 @@ def docgen_func(function_code, min_length, max_length, top_k, top_p, temp, repet
     }
     output = query(req_data)
     if type(output) is list:
-        return output[0]["generated_text"]
-        # return f'"""\n{" ".join(output[0]["generated_text"].split("\n"))}\n"""'
+        return f'"""\n{output[0]["generated_text"]}\n"""'
     else:
         msg = str(output)
         if msg == "{'error': 'Model stmnk/codet5-small-code-summarization-python is currently loading', 'estimated_time': 20}":