Spaces:
Running
Running
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 |
|