Update app.py

app.py

@@ -176,14 +176,19 @@ def overlay(image, mask, color, alpha, resize=None):
      
 
 
+model_path = 'models'
+general_model_name = 'e50_aug.pt'
+image_model_name = 'e100_img.pt'
 
-general_model_path = 'e50_aug.pt'
-image_model_path = 'e100_img.pt'
+general_model = YOLO(os.path.join(model_path, general_model_name))
+image_model = YOLO(os.path.join(model_path, image_model_name))
 
-general_model = YOLO(general_model_path)
-image_model = YOLO(image_model_path)
+image_path = 'examples'
+sample_name = ['0040da34-25c8-4a5a-a6aa-36733ea3b8eb.png', '00fb93d5-7c67-4851-ad08-f23ed2159467.png',
+               '0050a8ee-382b-447e-9c5b-8506d9507bef.png', '0064d3e2-3ba2-4332-a28f-3a165f2b84b1.png',
+               '019384d0-88c2-46ba-8f1b-bf7432f50ea3.png']
 
-sample_path = ['0040da34-25c8-4a5a-a6aa-36733ea3b8eb.png']
+sample_path = [os.path.join(image_path, sample) for sample in sample_name]
 
 flags = {
     'hist': False,