Update config.py

config.py

@@ -13,11 +13,11 @@ config = {
     'db_constructor_path': 'lib_to_constructor.csv',
 
     # retrieval_model_setting
-    'max_k': 2,
+    'max_k': 1,
 
     # generative_model_setting
-    'num_beams': 2,
-    'num_return_sequences': 2,
+    'num_beams': 1,
+    'num_return_sequences': 1,
 
     # hw_classifier setting
     'max_length': 128