From 1798324285a37d04102e8198a701bb23a0d6af9a Mon Sep 17 00:00:00 2001
From: Gaehwan Jo <gaehwan.jo@stradvision.com>
Date: Tue, 30 Apr 2024 02:28:27 +0000
Subject: [PATCH] error code fix

---
 infer.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/infer.py b/infer.py
index ce29670..45b6689 100644
--- a/infer.py
+++ b/infer.py
@@ -9,7 +9,7 @@
 def parse_args():
     parser = ArgumentParser()
 
-    parser.add_argument('model-path', type=str,
+    parser.add_argument('model_path', type=str,
                         help='path to checkpoint file containing trained model')
     parser.add_argument('-g', '--gpu', type=int, default=0,
                         help='gpu to use for inference (-1 for cpu)')