mutimodel5

app.py

@@ -11,12 +11,12 @@ def run_inference(input_image):
     
     input_image.save(input_image_path)  # Save the input image to disk
     
-    cmd = ['python', 'multi-model-ship-detection/yolov5-master/detect.py',
-           '--source', input_image_path, '--weights', 'multi-model-ship-detection/yolov5m-multi-model.pt', '--name', 'detection']
+    cmd = ['python', 'yolov5-master/detect.py',
+           '--source', input_image_path, '--weights', 'yolov5m-multi-model.pt', '--name', 'detection']
     
     subprocess.run(cmd)
     
-    folders = glob.glob('multi-model-ship-detection/yolov5-master/runs/detect/*')
+    folders = glob.glob('yolov5-master/runs/detect/*')
     folders = max(folders, key=os.path.getctime)
     output_image_path = glob.glob('{}/*'.format(folders))[0]
     output_image = Image.open(output_image_path)