diff --git a/Models/model_host.py b/Models/model_host.py
index 9febfec..68d5f45 100644
--- a/Models/model_host.py
+++ b/Models/model_host.py
@@ -157,12 +157,16 @@ def process_image():
 
 if __name__ == '__main__':
     path = os.getcwd()
-    resnet50_model_name='Weights/resnet50.h5.keras'
+    if(not path.endswith('Models')):
+      os.chdir(path + '/Weights')
+    path=os.getcwd()
+    print('Current directory is ..{current_directory}'.format(current_directory=path))
+    resnet50_model_name='../Weights/resnet50.h5.keras'
     resnet50_model_name = os.path.realpath(path + '/' +resnet50_model_name)
     print('Loading {model_name}'.format(model_name=resnet50_model_name))
     resnet50_model = keras.models.load_model(resnet50_model_name)  
 
-    resnet50_20241024_270_model_name='Weights/resnet50_20241024_270.h5.keras'
+    resnet50_20241024_270_model_name='../Weights/resnet50_20241024_270.h5.keras'
     resnet50_20241024_270_model_name = os.path.realpath(path + '/' + resnet50_20241024_270_model_name)
     print('Loading {model_name}'.format(model_name=resnet50_20241024_270_model_name))
     resnet50_20241024_270_model = keras.models.load_model(resnet50_20241024_270_model_name)
@@ -171,12 +175,12 @@ if __name__ == '__main__':
     # print('Loading {model_name}'.format(model_name=resnet50b_model_name))
     # resnet50b_model = keras.models.load_model(resnet50b_model_name)     
 
-    vgg16_model_name='Weights/vggnet16.h5.keras'
+    vgg16_model_name='../Weights/vggnet16.h5.keras'
     vgg16_model_name = os.path.realpath(path + '/' + vgg16_model_name)
     print('Loading {model_name}'.format(model_name=vgg16_model_name))    
     vgg16_model=keras.models.load_model(vgg16_model_name) 
 
-    lenet_model_name='Weights/lenet5.h5.keras'
+    lenet_model_name='../Weights/lenet5.h5.keras'
     lenet_model_name = os.path.realpath(path + '/' +lenet_model_name)
     print('Loading {model_name}'.format(model_name=lenet_model_name))
     lenet_model=keras.models.load_model(lenet_model_name)