Michael Rey commited on
Commit
7654541
·
1 Parent(s): 8e3a2ae

fixing problems

Browse files
Files changed (2) hide show
  1. src/monkey_resnet.pth +3 -0
  2. src/streamlit_app.py +1 -1
src/monkey_resnet.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:c7a3904aebf7b0ec02a2f7a389374de6ec062420e46ae087214b9c862ef66fd6
3
+ size 44805963
src/streamlit_app.py CHANGED
@@ -15,7 +15,7 @@ sys.path.append(os.path.abspath(os.path.join(os.path.dirname(__file__), "..")))
15
  st.set_page_config(page_title="ApexID: Monkey Species Classifier", layout="wide")
16
 
17
  # Constants for model path and class labels
18
- MODEL_PATH = "models/monkey_resnet.pth"
19
  CLASS_NAMES = ['n0', 'n1', 'n2', 'n3', 'n4', 'n5', 'n6', 'n7', 'n8', 'n9']
20
  LABEL_MAP = {
21
  'n0': 'alouatta_palliata',
 
15
  st.set_page_config(page_title="ApexID: Monkey Species Classifier", layout="wide")
16
 
17
  # Constants for model path and class labels
18
+ MODEL_PATH = "monkey_resnet.pth"
19
  CLASS_NAMES = ['n0', 'n1', 'n2', 'n3', 'n4', 'n5', 'n6', 'n7', 'n8', 'n9']
20
  LABEL_MAP = {
21
  'n0': 'alouatta_palliata',