From 1c5a70349db4fcf735916a1e63eb96b5b1666264 Mon Sep 17 00:00:00 2001
From: Krystian Wasilewski <krywas@st.amu.edu.pl>
Date: Fri, 9 Jun 2023 21:25:38 +0200
Subject: [PATCH] 122

---
 zad122.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/zad122.py b/zad122.py
index 6f021bf..5a58ab9 100644
--- a/zad122.py
+++ b/zad122.py
@@ -44,6 +44,9 @@ for folder_name in ['dev-0', 'test-A']:
                 if token in ['ia', 'ix', 'io', 'ik', 'ing']:
                     continue
 
+                # Biore pierwsze slowo z prawego kontekstu i sprawdzam czy jest jednym z tokenów przewidzianych
+                # przez prompt złożony z lewego kontekstu i kandydata na słowo w dziurze
+                # jesli tak to zwiększam prawdopodobieństwo tego slowa
                 new_prompt = f'{prefix} {token} '
                 new_inputs = tokenizer.encode(new_prompt, return_tensors="pt", truncation=True).to(device)
                 new_output = model(new_inputs)