Spaces:
Sleeping
Sleeping
fix: remove unused argument
Browse files- src/classifier/classifier_image.py +1 -2
- src/main.py +1 -1
src/classifier/classifier_image.py
CHANGED
|
@@ -11,7 +11,7 @@ from hf_push_observations import push_observations
|
|
| 11 |
from utils.grid_maker import gridder
|
| 12 |
from utils.metadata_handler import metadata2md
|
| 13 |
|
| 14 |
-
def cetacean_classify(cetacean_classifier
|
| 15 |
files = st.session_state.files
|
| 16 |
images = st.session_state.images
|
| 17 |
observations = st.session_state.observations
|
|
@@ -62,7 +62,6 @@ def cetacean_classify(cetacean_classifier, tab_inference):
|
|
| 62 |
|
| 63 |
whale_classes = out['predictions'][:]
|
| 64 |
# render images for the top 3 (that is what the model api returns)
|
| 65 |
-
#with tab_inference:
|
| 66 |
st.markdown(f"Top 3 Predictions for {file.name}")
|
| 67 |
for i in range(len(whale_classes)):
|
| 68 |
viewer.display_whale(whale_classes, i)
|
|
|
|
| 11 |
from utils.grid_maker import gridder
|
| 12 |
from utils.metadata_handler import metadata2md
|
| 13 |
|
| 14 |
+
def cetacean_classify(cetacean_classifier):
|
| 15 |
files = st.session_state.files
|
| 16 |
images = st.session_state.images
|
| 17 |
observations = st.session_state.observations
|
|
|
|
| 62 |
|
| 63 |
whale_classes = out['predictions'][:]
|
| 64 |
# render images for the top 3 (that is what the model api returns)
|
|
|
|
| 65 |
st.markdown(f"Top 3 Predictions for {file.name}")
|
| 66 |
for i in range(len(whale_classes)):
|
| 67 |
viewer.display_whale(whale_classes, i)
|
src/main.py
CHANGED
|
@@ -203,7 +203,7 @@ def main() -> None:
|
|
| 203 |
# TODO: cleaner design to disable the button until data input done?
|
| 204 |
st.info("Please upload an image first.")
|
| 205 |
else:
|
| 206 |
-
cetacean_classify(cetacean_classifier
|
| 207 |
|
| 208 |
|
| 209 |
|
|
|
|
| 203 |
# TODO: cleaner design to disable the button until data input done?
|
| 204 |
st.info("Please upload an image first.")
|
| 205 |
else:
|
| 206 |
+
cetacean_classify(cetacean_classifier)
|
| 207 |
|
| 208 |
|
| 209 |
|