Update model.py

src/models/model.py

@@ -4,3 +4,4 @@ import numpy as np
 import pandas as pd
 from datasets import load_metric
 from transformers import (
+    AdamW,