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']})")
|