Skip to content

Commit 2ca4cdd

Browse files
committed
Update run_pretrained_models.py
1 parent 3b5a381 commit 2ca4cdd

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

tests/run_pretrained_models.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -51,7 +51,8 @@
5151
def get_beach(shape):
5252
"""Get beach image as input."""
5353
resize_to = shape[1:3]
54-
path = os.path.join(os.path.dirname(os.path.abspath(__file__)), "beach.jpg")
54+
path = os.path.join(os.path.dirname(os.path.abspath(__file__)),
55+
'..', 'tests', "beach.jpg")
5556
img = PIL.Image.open(path)
5657
img = img.resize(resize_to, PIL.Image.ANTIALIAS)
5758
img_np = np.array(img).astype(np.float32)

0 commit comments

Comments
 (0)