vancauwe commited on
Commit
a874233
·
1 Parent(s): a62a456

fix: image path in whale viewer

Browse files
Files changed (1) hide show
  1. src/whale_viewer.py +4 -1
src/whale_viewer.py CHANGED
@@ -2,6 +2,7 @@ from typing import List
2
 
3
  from PIL import Image
4
  import pandas as pd
 
5
 
6
  WHALE_CLASSES = [
7
  "beluga",
@@ -149,7 +150,9 @@ def display_whale(whale_classes:List[str], i:int, viewcontainer=None):
149
  viewcontainer.markdown(
150
  "### :whale: #" + str(i + 1) + ": " + format_whale_name(whale_classes[i])
151
  )
152
- image = Image.open("images/references/" + df_whale_img_ref.loc[whale_classes[i], "WHALE_IMAGES"])
 
 
153
 
154
  viewcontainer.image(image, caption=df_whale_img_ref.loc[whale_classes[i], "WHALE_REFERENCES"])
155
  # link st.markdown(f"[{df.loc[whale_classes[i], 'WHALE_REFERENCES']}]({df.loc[whale_classes[i], 'WHALE_REFERENCES']})")
 
2
 
3
  from PIL import Image
4
  import pandas as pd
5
+ import os
6
 
7
  WHALE_CLASSES = [
8
  "beluga",
 
150
  viewcontainer.markdown(
151
  "### :whale: #" + str(i + 1) + ": " + format_whale_name(whale_classes[i])
152
  )
153
+ current_dir = os.getcwd()
154
+ image_path = os.path.join(current_dir, "src/images/references/")
155
+ image = Image.open(image_path + df_whale_img_ref.loc[whale_classes[i], "WHALE_IMAGES"])
156
 
157
  viewcontainer.image(image, caption=df_whale_img_ref.loc[whale_classes[i], "WHALE_REFERENCES"])
158
  # link st.markdown(f"[{df.loc[whale_classes[i], 'WHALE_REFERENCES']}]({df.loc[whale_classes[i], 'WHALE_REFERENCES']})")