Add pipeline tag to metadata

README.md

@@ -1,7 +1,8 @@
 ---
-library_name: peft
 base_model: WizardLMTeam/WizardCoder-Python-13B-V1.0
+library_name: peft
 license: apache-2.0
+pipeline_tag: text-generation
 ---
 
 # Model Card for Model ID
@@ -39,17 +40,23 @@ def extract_diff(model_output):
     pattern = re.compile(r'\s*\[DIFF\](.*?)\[/DIFF\]\s*', re.DOTALL)
     matches = pattern.findall(model_output)
     if matches:
-        return matches[0].strip('\n')
+        return matches[0].strip('
+')
     return None
 
 def diff2code(diff : str) -> str:
-    return '\n'.join(
+    return '
+'.join(
         line[2:] for line in diff.splitlines()
         if not line.startswith('-')
     ).strip()
 
 def corrupt(program, model, tokenizer, **generation_kwargs):
-    prompt = f'[PYTHON]\n{program.strip()}\n[/PYTHON]\n[DIFF]\n'
+    prompt = f'[PYTHON]
+{program.strip()}
+[/PYTHON]
+[DIFF]
+'
     model_inputs = tokenizer([prompt], return_tensors="pt").to(model.device)
     generated_ids = model.generate(**model_inputs, **generation_kwargs)
     outputs = tokenizer.batch_decode(generated_ids, skip_special_tokens=True)