Update app.py

app.py

@@ -185,7 +185,8 @@ 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 f'"""\n{" || ".join(output[0]["generated_text"].split("|"))}\n"""'
+        return output[0]["generated_text"]
+        # return f'"""\n{" ".join(output[0]["generated_text"].split("\n"))}\n"""'
     else:
         msg = str(output)
         if msg == "{'error': 'Model stmnk/codet5-small-code-summarization-python is currently loading', 'estimated_time': 20}":