test

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)