Spaces:
Sleeping
Sleeping
Michael Rey
commited on
Commit
·
99d4d0e
1
Parent(s):
d5590ac
fixed file path issue
Browse files- src/streamlit_app.py +1 -1
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 = "
|
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 = os.path.join(os.path.dirname(__file__), "monkey_resnet.pth")
|
19 |
CLASS_NAMES = ['n0', 'n1', 'n2', 'n3', 'n4', 'n5', 'n6', 'n7', 'n8', 'n9']
|
20 |
LABEL_MAP = {
|
21 |
'n0': 'Alouatta Palliata',
|