mohdelgaar commited on
Commit
2ac88c4
·
1 Parent(s): 32702e0

remove compute_lng

Browse files
Files changed (3) hide show
  1. app.py +5 -5
  2. model.py +0 -1
  3. requirements.txt +2 -8
app.py CHANGED
@@ -1,7 +1,8 @@
1
- import spacy
2
- import nltk
3
- nltk.download('wordnet', quiet=True)
4
- spacy.cli.download('en_core_web_sm')
 
5
 
6
  import torch
7
  import joblib, json
@@ -12,7 +13,6 @@ from const import used_indices, name_map
12
  from model import get_model
13
  from options import parse_args
14
  from transformers import T5Tokenizer
15
- from compute_lng import compute_lng
16
  from sklearn.experimental import enable_iterative_imputer
17
  from sklearn.impute import IterativeImputer
18
  from sklearn.linear_model import Ridge
 
1
+ # import spacy
2
+ # import nltk
3
+ # nltk.download('wordnet', quiet=True)
4
+ # spacy.cli.download('en_core_web_sm')
5
+ # from compute_lng import compute_lng
6
 
7
  import torch
8
  import joblib, json
 
13
  from model import get_model
14
  from options import parse_args
15
  from transformers import T5Tokenizer
 
16
  from sklearn.experimental import enable_iterative_imputer
17
  from sklearn.impute import IterativeImputer
18
  from sklearn.linear_model import Ridge
model.py CHANGED
@@ -5,7 +5,6 @@ import numpy as np
5
  from torch import nn
6
  from transformers import T5ForConditionalGeneration, T5EncoderModel, AutoModel, LogitsProcessor, LogitsProcessorList
7
  from functools import partial
8
- from compute_lng import compute_lng
9
  from undecorate import unwrap
10
  from types import MethodType
11
  from utils import *
 
5
  from torch import nn
6
  from transformers import T5ForConditionalGeneration, T5EncoderModel, AutoModel, LogitsProcessor, LogitsProcessorList
7
  from functools import partial
 
8
  from undecorate import unwrap
9
  from types import MethodType
10
  from utils import *
requirements.txt CHANGED
@@ -1,13 +1,7 @@
1
  torch
2
- torchvision
3
  numpy
4
  joblib
5
- gensim
6
- supar
7
  transformers
8
  scikit-learn
9
- tqdm
10
- spacy
11
- sentencepiece
12
- lftk
13
- undecorate
 
1
  torch
 
2
  numpy
3
  joblib
4
+ undecorate
 
5
  transformers
6
  scikit-learn
7
+ tqdm