diff --git a/test.py b/test.py index 2e11130..2b0102f 100644 --- a/test.py +++ b/test.py @@ -17,6 +17,6 @@ def generate_response(prompt, max_length=100): response = tokenizer.decode(outputs[0], skip_special_tokens=True) return response -prompt = "Ile dni urlopu przysługuje pracownikowi z 5 letnim stazem pracy w pełnym wymiarze pracy?" +prompt = "Zacytuj art. 154 kodeksu pracy" response = generate_response(prompt) print(response) \ No newline at end of file