log: test path

model.py

@@ -2,6 +2,7 @@ import torch
 import torch.nn as nn
 import wandb
 import streamlit as st
+import os
 
 import clip
 from transformers import GPT2Tokenizer, GPT2LMHeadModel
@@ -128,8 +129,9 @@ def load_model():
     gpt_model, tokenizer = load_gpt_model()
    
     
-
     # Load weights
+    print(PATH)
+    print(os.getcwd())
     model = ImageCaptioner(clip_model, gpt_model, tokenizer, 0)
     checkpoint = torch.load(PATH, map_location=torch.device('cpu'))
     model.load_state_dict(checkpoint["state_dict"])