diff --git a/hft.py b/hft.py index 606933a..97698af 100644 --- a/hft.py +++ b/hft.py @@ -219,9 +219,7 @@ def generate_answer_with_source(question, model, tokenizer, source_mapper, max_l # Testowanie test_questions = [ - "Jaki jest wymiar urlopu wypoczynkowego?", - "Jakie są zasady bezpieczeństwa na budowie?", - "Wyjaśnij procedurę zwolnienia grupowego" + "Jak brzmi art. 154 kodeksu pracy?" ] print("\n=== TEST MODELU ===")