Spaces:
Sleeping
Sleeping
fix: image path in whale viewer
Browse files- 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 |
-
|
|
|
|
|
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']})")
|