diff --git a/1/model.py b/1/model.py
index 8556bba..6c6acbf 100644
--- a/1/model.py
+++ b/1/model.py
@@ -3,6 +3,7 @@ import torch
 import numpy as np
 import triton_python_backend_utils as pb_utils
 import uuid
+import transformers
 from typing import List, Dict, Any, Union, Tuple
 
 from transformers import (