Update model.py

model.py

@@ -131,7 +131,10 @@ def load_model():
     
     # Load weights
     print(PATH)
-    print(os.getcwd())
+
+    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"])