check whether custom_model.pt exists

sample_solution.py

@@ -264,6 +264,7 @@ def main(image):
     # print(image.shape)
     # print()
     # cv2.imshow("sdfsd", image)
+    print("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)