added inference code

README.md

@@ -9,4 +9,37 @@ tags:
 - fine-tune
 - salary-normalizer
 - salary-parser
----
+---
+
+# Implementation Code
+
+```
+from transformers import AutoTokenizer, AutoModelForCausalLM
+
+model_name = "Draup/salary-normalizer"
+
+tokenizer = AutoTokenizer.from_pretrained(model_name)
+model = AutoModelForCausalLM.from_pretrained(model_name)
+
+salary_text = "12 to 12.5 US $ per hr"
+country = "United States"
+
+prompt = (
+    f"<start_of_turn>user\n"
+    f"summarize salary: {salary_text}\n"
+    f"country: {country}<end_of_turn>\n"
+    f"<start_of_turn>model\n"
+)
+
+inputs = tokenizer(prompt, return_tensors="pt", truncation=True)
+outputs = model.generate(
+    **inputs,
+    max_new_tokens=70,
+    do_sample=False,
+    eos_token_id=tokenizer.convert_tokens_to_ids("<end_of_turn>")
+)
+
+result = tokenizer.decode(outputs[0][inputs["input_ids"].shape[-1]:], skip_special_tokens=True)
+print(result)
+# Output: {"currency": "US $", "iso_code": "USD", "min_amount": 12, "max_amount": 12.5, "pay_rate": "HOURLY"}
+```