diff --git a/biblioteka_DL b/biblioteka_DL
index d3cdfc2..1397718 100644
--- a/biblioteka_DL
+++ b/biblioteka_DL
@@ -73,4 +73,6 @@ with torch.no_grad():
     acc = y_predicted_cls.eq(y_testing).sum()/float(y_testing.shape[0])
     print(f'{acc:.4f}')
     result = open("output",'w+')
-    result.write(f'{y_predicted}')
\ No newline at end of file
+    result.write(f'{y_predicted}')
+
+torch.save(model, "model.pkl")
\ No newline at end of file