Michael Rey commited on
Commit
6abf0ea
·
1 Parent(s): eb6a8a0

modified streamlit code

Browse files
Files changed (1) hide show
  1. src/streamlit_app.py +1 -1
src/streamlit_app.py CHANGED
@@ -5,7 +5,7 @@ import torch
5
  import torchvision.transforms as transforms
6
  from PIL import Image
7
 
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
 
5
  import torchvision.transforms as transforms
6
  from PIL import Image
7
 
8
+ from 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