Michael Rey commited on
Commit
4d6d4c8
·
1 Parent(s): 756ef40

modified streamlit_app.py

Browse files
Files changed (2) hide show
  1. src/__init__.py +0 -0
  2. src/streamlit_app.py +3 -2
src/__init__.py DELETED
File without changes
src/streamlit_app.py CHANGED
@@ -1,7 +1,5 @@
1
  import sys
2
  import os
3
- sys.path.append(os.path.abspath(os.path.join(os.path.dirname(__file__), "..")))
4
-
5
  import streamlit as st
6
  import torch
7
  import torchvision.transforms as transforms
@@ -10,6 +8,9 @@ from PIL import Image
10
  from models.resnet_model import MonkeyResNet
11
  from utils.data_loader import get_data_loaders
12
 
 
 
 
13
  # Set Streamlit page configuration
14
  st.set_page_config(page_title="ApexID: Monkey Species Classifier", layout="wide")
15
 
 
1
  import sys
2
  import os
 
 
3
  import streamlit as st
4
  import torch
5
  import torchvision.transforms as transforms
 
8
  from models.resnet_model import MonkeyResNet
9
  from utils.data_loader import get_data_loaders
10
 
11
+ # Ensure the parent directory is in the system path for module imports
12
+ sys.path.append(os.path.abspath(os.path.join(os.path.dirname(__file__), "..")))
13
+
14
  # Set Streamlit page configuration
15
  st.set_page_config(page_title="ApexID: Monkey Species Classifier", layout="wide")
16