Salimshakeel commited on
Commit
a4d2baa
·
1 Parent(s): 66de988
Files changed (2) hide show
  1. Dockerfile +2 -0
  2. services/model_loader.py +1 -1
Dockerfile CHANGED
@@ -3,6 +3,8 @@
3
 
4
  FROM python:3.12-slim
5
 
 
 
6
  WORKDIR /code
7
 
8
  COPY ./requirements.txt /code/requirements.txt
 
3
 
4
  FROM python:3.12-slim
5
 
6
+ ENV TORCH_HOME=/tmp/torch_cache
7
+
8
  WORKDIR /code
9
 
10
  COPY ./requirements.txt /code/requirements.txt
services/model_loader.py CHANGED
@@ -15,5 +15,5 @@ def load_model(weights_path):
15
  pos_enc="absolute"
16
  ).to(DEVICE)
17
  # model.load_state_dict(torch.load(weights_path, map_location=DEVICE))
18
- model.eval
19
  return model
 
15
  pos_enc="absolute"
16
  ).to(DEVICE)
17
  # model.load_state_dict(torch.load(weights_path, map_location=DEVICE))
18
+ model.eval()
19
  return model