vancauwe commited on
Commit
d8c87e7
·
1 Parent(s): 0e8c927

fix: remove unused argument

Browse files
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, tab_inference):
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, tab_inference)
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