Update README.md
Browse files
README.md
CHANGED
|
@@ -75,7 +75,8 @@ fine_tuned_prompt = """
|
|
| 75 |
### Response:
|
| 76 |
{}"""
|
| 77 |
|
| 78 |
-
|
|
|
|
| 79 |
inputs = tokenizer(model_input, return_tensors="pt").to("cuda")
|
| 80 |
outputs = model.generate(**inputs, max_new_tokens=256, temperature=1.0)
|
| 81 |
output = tokenizer.decode(outputs[0], skip_special_tokens=True)
|
|
|
|
| 75 |
### Response:
|
| 76 |
{}"""
|
| 77 |
|
| 78 |
+
example_input="Many of ZAR’s grounds of appeal related to fact finding. Drawing on principles set down in several other courts and tribunals, the Appeal Panel summarised the circumstances in which leave may be granted for a person to appeal from findings of fact: <CASENAME> at [84]."
|
| 79 |
+
model_input = fine_tuned_prompt.format("Predict the name of the case that needs to be cited in the text and explain why it should be cited.", example_input, '')
|
| 80 |
inputs = tokenizer(model_input, return_tensors="pt").to("cuda")
|
| 81 |
outputs = model.generate(**inputs, max_new_tokens=256, temperature=1.0)
|
| 82 |
output = tokenizer.decode(outputs[0], skip_special_tokens=True)
|