0

requirements.txt

@@ -2,4 +2,5 @@ streamlit
 tensorflow
 keras
 numpy
-matplotlib
+matplotlib
+os

src/app.py

@@ -2,9 +2,12 @@ import streamlit as st
 from keras.models import load_model
 from keras.preprocessing.image import load_img, img_to_array
 import numpy as np
+import os
 
 # Load model
-model = load_model('FishModel_VGG16.h5', compile=False)
+model_path = os.path.join(os.path.dirname(__file__), 'FishModel_VGG16.h5')
+model = load_model(model_path, compile=False)
+
 
 # Class labels
 class_names = ['Bangus', 'Big Head Carp', 'Black Spotted Barb', 'Catfish', 'Climbing Perch',