diff --git a/allegro.py b/allegro.py
index 6eef052..39e8018 100644
--- a/allegro.py
+++ b/allegro.py
@@ -111,8 +111,8 @@ def main():
 
     print("Rozpoczęcie treningu...")
     trainer.train()
-    trainer.save_model("./trained_model/gpt")
-    tokenizer.save_pretrained("./trained_model/gpt")
+    trainer.save_model("./trained_model/allegro")
+    tokenizer.save_pretrained("./trained_model/allegro")
 
 if __name__ == "__main__":
     main()
\ No newline at end of file