dodanie przykładu uzycia na koniec

2025-02-25 17:35:29 +01:00 · 2025-02-25 17:35:29 +01:00 · 1a41ca184b
parent 414805d574
commit 1a41ca184b
1 changed files with 5 additions and 0 deletions
--- a/hft.py
+++ b/hft.py
@ -205,3 +205,8 @@ def generate_answer(question, model, tokenizer, source_mapper, max_length=200):
    # Pobierz źródło z ostatniego tokena
    last_token_id = outputs.sequences[0][-1].item()
    source_idx = model.source_embeddi
+
+# Przykład użycia
+question = "Ile dni urlopu przysługuje pracownikowi?"
+answer = generate_answer(question, model, tokenizer, source_mapper)
+print(answer)