Changed javadoc prompt

app.py

@@ -69,10 +69,11 @@ def generate_code(task_type: str, nl_input: str):
         # return single_function(response)
         return _clean_csharp_output(response)
     elif task_type=="Generate Java Docs":
-        prompt= f"### Instruction:\n\nGenerate Java Docs for the below Java Code\n\n{nl_input}\n\n### Response:\n\n"
+        prompt= f"### Instruction:\n\Write a Javadoc comment (starting with /** and ending with */) for the following Java 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=128, 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()
+        response = response.split("*/")[0] + "*/" if "*/" in response else response
         return response
 
 demo = gr.Interface(