Updated prompt for document generation with correct indentation again

app.py

@@ -83,6 +83,7 @@ def generate_code(task_type: str, nl_input: str):
     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/**"
+        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()
         if not response.startswith("/**"):