check whether custom_model.pt exists

sample_solution.py

@@ -264,7 +264,8 @@ def main(image):
     # print(image.shape)
     # print()
     # cv2.imshow("sdfsd", image)
-    print("custom_model.pt" in os.listdir("./"))
+    print("List of files: ", os.listdir("./"))
+    print("Model exists?: ", "custom_model.pt" in os.listdir("./"))
     image_path = os.path.join("images", str(time.time()) + ".jpg")
     image = cv2.cvtColor(image, cv2.COLOR_RGB2BGR)
     cv2.imwrite(image_path, image)