wracell commited on
Commit
3e6b952
·
1 Parent(s): 652d7e3
Files changed (3) hide show
  1. Dockerfile +1 -0
  2. requirements.txt +1 -2
  3. src/app.py +1 -3
Dockerfile CHANGED
@@ -11,6 +11,7 @@ RUN apt-get update && apt-get install -y \
11
 
12
  COPY requirements.txt ./
13
  COPY src/ ./src/
 
14
 
15
  RUN pip3 install -r requirements.txt
16
 
 
11
 
12
  COPY requirements.txt ./
13
  COPY src/ ./src/
14
+ COPY src/ .
15
 
16
  RUN pip3 install -r requirements.txt
17
 
requirements.txt CHANGED
@@ -2,5 +2,4 @@ streamlit
2
  tensorflow
3
  keras
4
  numpy
5
- matplotlib
6
- os
 
2
  tensorflow
3
  keras
4
  numpy
5
+ matplotlib
 
src/app.py CHANGED
@@ -5,9 +5,7 @@ import numpy as np
5
  import os
6
 
7
  # Load model
8
- model_path = os.path.join(os.path.dirname(__file__), 'FishModel_VGG16.h5')
9
- model = load_model(model_path, compile=False)
10
-
11
 
12
  # Class labels
13
  class_names = ['Bangus', 'Big Head Carp', 'Black Spotted Barb', 'Catfish', 'Climbing Perch',
 
5
  import os
6
 
7
  # Load model
8
+ model = load_model('FishModel_VGG16.h5', compile=False)
 
 
9
 
10
  # Class labels
11
  class_names = ['Bangus', 'Big Head Carp', 'Black Spotted Barb', 'Catfish', 'Climbing Perch',