diff --git a/hft.py b/hft.py
index 78d73f1..fa94e42 100644
--- a/hft.py
+++ b/hft.py
@@ -151,7 +151,7 @@ class CustomTrainer(Trainer):
 
 # Inicjalizacja komponentów
 source_mapper = SourceMapper()
-model_name = "google/gemma-2-2b"
+model_name = "perplexity-ai/r1-1776"#"google/gemma-2-2b"
 tokenizer = AutoTokenizer.from_pretrained(model_name)
 tokenizer.pad_token = tokenizer.eos_token