From 2fb7404168508052876f1fe783b49dfbc475a4c6 Mon Sep 17 00:00:00 2001
From: Gabriel Dutra <57206480+Dutra-Apex@users.noreply.github.com>
Date: Tue, 14 Feb 2023 15:52:49 -0500
Subject: [PATCH] Fixed predict.py resulting in 'output_paths' error

---
 predict.py | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/predict.py b/predict.py
index 46ff358..526e054 100644
--- a/predict.py
+++ b/predict.py
@@ -9,6 +9,7 @@
 from config import main_config
 from models import gmcnn_gan
 from utils import training_utils
+from utils import constants
 
 log = training_utils.get_logger()
 
@@ -46,6 +47,10 @@ def main():
                       required=True,
                       help='The path to the mask')
   
+  parser.add_argument('--experiment_name',
+                      required=True,
+                      help='Name of the experiment to load weights from')
+  
   parser.add_argument('--save_to',
                       default='predicted.jpg',
                       help='The save path of predicted image')
@@ -59,6 +64,7 @@ def main():
                                    img_width=config.training.img_width,
                                    num_channels=config.training.num_channels,
                                    warm_up_generator=False,
+                                   output_paths=constants.OutputPaths(args.experiment_name),
                                    config=config)
   log.info('Loading GMCNN model...')
   gmcnn_model.load()