Added model

example-dummy-evaluation.pkl

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:db681019f6fde303497efa1d273f12002c2ae41ce973615e81b3f1d246a368e7
+size 1912

train.py

@@ -1,3 +1,4 @@
+import pickle
 from transformers import PreTrainedModel, PretrainedConfig
 
 class CountAModel(PreTrainedModel):
@@ -22,5 +23,9 @@ sentence = "This is a sample sentence with a few 'a's."
 count_a = model(sentence)
 print(f"The sentence contains {count_a} letter(s) 'a'.")
 
+
 # Save the model in the current directory
 model.save_pretrained(".")
+with open('example-dummy-evaluation.pkl', 'wb') as f:
+    pickle.dump(model, f)
+print("Model saved successfully.")