Update config.py

config.py

@@ -13,14 +13,14 @@ config = {
     'db_constructor_path': 'lib_to_constructor.csv',
 
     # retrieval_model_setting
-    'max_k': 10,
+    'max_k': 2,
 
     # generative_model_setting
-    'num_beams': 5,
-    'num_return_sequences': 3,
+    'num_beams': 2,
+    'num_return_sequences': 2,
 
     # hw_classifier setting
-    'max_length': 512
+    'max_length': 128
 }
 
 classifier_class_mapping = {