Hugging Face's logo

Spaces:
SantiagoMoreno-UdeA
/
NER_RC
Runtime error

App Files Community
NER_RC
File size: 3,499 Bytes 
42d6a0f
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
 
#!/bin/bash
# NER software handler

if [ $# -gt 0 ]
    then
    MODE="$1"
    STANDARD="False"
    FAST="False"
    CUDA="False"
    UFLAG="False"
        if [ ${MODE} == 'TRAIN' ]
        then
            shift # past argument
            if [ $# -gt 1 ]
                then
                while [[ $# -gt 1 ]]; do
                    case $1 in
                        -f|--fast)
                        FAST="$2"
                        shift # past argument
                        shift # past value
                        ;;

                        -m|--model)
                        MODEL="$2"
                        shift # past argument
                        shift # past value
                        ;;

                        -s|--standard)
                        STANDARD="$2"
                        shift # past argument
                        shift # past value
                        ;;

                        -id|--inputdir)
                        INPUTDIR="$2"
                        shift # past argument
                        shift # past value
                        ;;

                        -u|--upsampleflag)
                        UFLAG="$2"
                        shift # past argument
                        shift # past value
                        ;;

                        -cu|--cuda)
                        CUDA="$2"
                        shift # past argument
                        shift # past value
                        ;;

                    esac
                done
                    python src/scripts/Train_model.py -f ${FAST} -m ${MODEL} -s ${STANDARD} -id "${INPUTDIR}" -u "${UFLAG}" -cu "${CUDA}"
            else
                echo Not arguments the script requires at least input directory
            fi


        elif [ $1 == 'USE' ]
        then
        shift # past argument
        if [ $# -gt 1 ]
            then
            while [[ $# -gt 1 ]]; do
                case $1 in
                    -m|--model)
                    MODEL="$2"
                    shift # past argument
                    shift # past value
                    ;;

                    -id|--inputdir)
                    INPUTDIR="$2"
                    shift # past argument
                    shift # past value
                    ;;

                    -od|--outputdir)
                    OUTPUTDIR="$2"
                    shift # past argument
                    shift # past value
                    ;;

                    -cu|--cuda)
                    CUDA="$2"
                    shift # past argument
                    shift # past value
                    ;;

                esac

            done
            if [ -n "${OUTPUTDIR}" ] && [ -n "${CUDA}" ]; then
                python src/scripts/Tagged_document.py -m ${MODEL} -id "${INPUTDIR}" -od "${OUTPUTDIR}" -cu "${CUDA}"

            elif [[ -n "${OUTPUTDIR}" ]]; then
                python src/scripts/Tagged_document.py -m ${MODEL} -id "${INPUTDIR}" -od "${OUTPUTDIR}" 

            elif [[ -n "${CUDA}" ]]; then
                python src/scripts/Tagged_document.py -m ${MODEL} -id "${INPUTDIR}" -cu "${CUDA}"

            else
                python src/scripts/Tagged_document.py -m ${MODEL} -id "${INPUTDIR}"
            fi
        

        else
            echo Not arguments the script requires at least model and input file
        fi

    else
        echo invalid option, USE for use a model, TRAIN for train a new model
    fi

fi