Changed prompt

app.py

@@ -47,7 +47,7 @@ def _clean_csharp_output(code: str) -> str:
 @spaces.GPU
 def generate_code(task_type: str, nl_input: str):
     if task_type == "NL to Java" or task_type == "End to End NL to C#":
-        prompt = f"### Instruction:\n\n{nl_input} Write the solution in Java.\n\n### Response:\n\n"
+        prompt = f"### Instruction:\n\n{nl_input} Write the solution in Java with proper indentation.\n\n### Response:\n\n"
         inputs = tokenizer(prompt, return_tensors="pt").to(model.device)
         outputs = model.generate(**inputs, max_new_tokens=128, eos_token_id=tokenizer.eos_token_id, pad_token_id=tokenizer.eos_token_id)
         response = tokenizer.decode(outputs[0], skip_special_tokens=True).split("### Response:\n\n")[-1].strip()
@@ -55,14 +55,14 @@ def generate_code(task_type: str, nl_input: str):
             # return single_function(response)
             return _clean_csharp_output(response)
         else:
-            prompt = f"### Instruction:\n\nConvert the below Java Code to C#\n\n{single_function(response)}\n\n### Response:\n\n"
+            prompt = f"### Instruction:\n\nConvert the below Java Code to C# with proper indentation\n\n{single_function(response)}\n\n### Response:\n\n"
             inputs = tokenizer(prompt, return_tensors="pt").to(model.device)
             outputs = model.generate(**inputs, max_new_tokens=128, eos_token_id=tokenizer.eos_token_id, pad_token_id=tokenizer.eos_token_id)
             response = tokenizer.decode(outputs[0], skip_special_tokens=True).split("### Response:\n\n")[-1].strip()
             # return single_function(response)
             return _clean_csharp_output(response)
     elif task_type == "Java to C#":
-        prompt = f"### Instruction:\n\nConvert the below Java Code to C#\n\n{nl_input}\n\n### Response:\n\n"
+        prompt = f"### Instruction:\n\nConvert the below Java Code to C# with proper indentation\n\n{nl_input}\n\n### Response:\n\n"
         inputs = tokenizer(prompt, return_tensors="pt").to(model.device)
         outputs = model.generate(**inputs, max_new_tokens=128, eos_token_id=tokenizer.eos_token_id, pad_token_id=tokenizer.eos_token_id)
         response = tokenizer.decode(outputs[0], skip_special_tokens=True).split("### Response:\n\n")[-1].strip()