fix: wrong path

Dockerfile

@@ -15,8 +15,6 @@ ENV HOME /home/user
 ENV PATH $HOME/.local/bin:$PATH
 
 WORKDIR $HOME
-RUN git lfs pull -I artifacts/model-ql03493w:v3/model.ckpt
-
 RUN mkdir app
 WORKDIR $HOME/app
 COPY . $HOME/app

model.py

@@ -122,7 +122,7 @@ def load_model():
 
     # # Load fine-tuned model from wandb
     artifact_dir = "./artifacts/model-ql03493w:v3"
-    PATH = f"{artifact_dir[2:]}/model.ckpt"
+    PATH = f"{os.getcwd()}/{artifact_dir[2:]}/model.ckpt"
 
     # Load pretrained GPT, CLIP model from OpenAI
     clip_model, image_transfrom = load_clip_model()
@@ -131,10 +131,6 @@ def load_model():
     
     # Load weights
     print(PATH)
-
-    for file_name in os.listdir(os.getcwd()):
-        print(file_name)
-        
     model = ImageCaptioner(clip_model, gpt_model, tokenizer, 0)
     checkpoint = torch.load(PATH, map_location=torch.device('cpu'))
     model.load_state_dict(checkpoint["state_dict"])