Updated prompt for document generation with multi line indentation

app.py

@@ -81,8 +81,8 @@ def generate_code(task_type: str, nl_input: str):
         output= _clean_csharp_output(response)
         return format_code(output, language="cs")
     elif task_type=="Generate Documentation":
-        # prompt= f"### Instruction:\n\nGenerate documentation for the following code:\n\n{nl_input}\n\n### Response:\n\n/**"
-        prompt = f"### Instruction:\n\nWrite a documentation comment with @param and @return tags for the following method:\n\n{nl_input}\n\n### Response:\n\n/**"
+        prompt = f"### Instruction:\n\nWrite a multi-line Javadoc comment with each tag on a new line, with @param and @return tags for the following method:\n\n{nl_input}\n\n### Response:\n\n/**"
+        # prompt = f"### Instruction:\n\nWrite a documentation comment with @param and @return tags for the following method:\n\n{nl_input}\n\n### Response:\n\n/**"
         inputs = tokenizer(prompt, return_tensors="pt").to(model.device)
         output=model.generate(**inputs,max_new_tokens=200, eos_token_id=tokenizer.eos_token_id, pad_token_id=tokenizer.eos_token_id)
         response = tokenizer.decode(output[0], skip_special_tokens=True).split("### Response:\n\n")[-1].strip()