Spaces:
Runtime error
Runtime error
| # -*- coding: utf-8 -*- | |
| """ | |
| Created on Tue Oct 25 00:32:02 2022 | |
| @author: Santiago Moreno | |
| """ | |
| import os | |
| import argparse | |
| from functions import use_model, str2bool, usage_cuda | |
| default_path = os.path.dirname(os.path.abspath(__file__)) | |
| os.chdir(default_path) | |
| output_dir = "../../data/tagged/document_tagged.json" | |
| if __name__ == '__main__': | |
| parser = argparse.ArgumentParser(add_help=True, usage='Tag a document with a pre-trained model (GPU optional)') | |
| parser.add_argument('-m','--model', default='CCC', type=str, nargs='?', help='New model name', required=True) | |
| parser.add_argument('-id','--input_data', type=str, nargs='?', help='Absolute path input file', required=True) | |
| parser.add_argument('-od','--output_data', const=output_dir, default=output_dir, type=str, nargs='?', help='Absolute path output file', required=False) | |
| parser.add_argument('-cu','--cuda', type=str2bool, nargs='?', const=True, default=False, help='Boolean value for using cuda to Train the model (True). By defaul False.', choices=(True, False), required=False) | |
| args = parser.parse_args() | |
| #print(args.model, args.input_data, args.output_data) | |
| if args.cuda: cuda_info = usage_cuda(True) | |
| else: cuda_info = usage_cuda(False) | |
| print(cuda_info) | |
| Error = use_model(args.model, args.input_data, args.output_data) | |
| if type(Error)==int: | |
| print('Tagged not complete, error code {}'.format(Error)) | |
| else: | |
| print('Tagged complete') | |
| # path_data = "C:/Users/gita/OneDrive - Universidad de Antioquia/GITA/Maestría/Programas/Datasets/camara_comercio_NER/gt/3cb4fa20-89cb-11e8-a485-d149999fe64b-0.json " | |
| # output_dir = "C:/Users/gita/OneDrive - Universidad de Antioquia/GITA/Maestría/Programas/Software NER/document_tagged.json" | |
| # sentence = use_model('CCC', path_data, output_dir) | |