From 714c481edcefe91f2c3a46d9aca6098a77cd085e Mon Sep 17 00:00:00 2001
From: weishan <weishan.wyf@alibaba-inc.com>
Date: Sat, 1 Jul 2023 22:47:26 +0800
Subject: [PATCH] fix: make query and database required

---
 src/dinov2_retrieval/__init__.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/src/dinov2_retrieval/__init__.py b/src/dinov2_retrieval/__init__.py
index 9fd180d..2fed2c7 100644
--- a/src/dinov2_retrieval/__init__.py
+++ b/src/dinov2_retrieval/__init__.py
@@ -31,12 +31,14 @@ def main():
         "-q",
         "--query",
         type=str,
+        required=True,
         help="path to a query image file or image folder",
     )
     parser.add_argument(
         "-d",
         "--database",
         type=str,
+        required=True,
         help="path to the database image file or image folder",
     )
     parser.add_argument(