first

pipeline.py

@@ -7,7 +7,7 @@ custokenizer = AutoTokenizer.from_pretrained("./GPT2/phoBert", use_fast=False)
 # -------Load model saved-----------------#
 
 parser = parser_args()
-parser.add_argument('--img_path', help='img path', default='./test_imgs/test.jpg')
+#parser.add_argument('--img_path', help='img path', default='./test_imgs/test.jpg')
 parser.add_argument('--config', help='config file', default='./query2labels/output/config.json')
 parser.add_argument('-f')
 args = parser.parse_args()
@@ -24,8 +24,8 @@ poem = pipeline('text-generation', model="./GPT2/rkw_4sen",
                 config=config)
 
 
-def main():
-    clses = vis_extractor.predict(args.img_path)
+def main(img_path):
+    clses = vis_extractor.predict(img_path)
     
     keywords = clses
     print(keywords)