alakxender commited on
Commit
0dddab9
·
1 Parent(s): b04ca6e
Files changed (1) hide show
  1. app.py +8 -1
app.py CHANGED
@@ -29,10 +29,17 @@ lm5gram = hf_hub_download(
29
  filename="language_model/5gram.bin",
30
  )
31
 
 
 
 
 
 
 
 
32
  lexicon_file = hf_hub_download(
33
  repo_type="dataset",
34
  repo_id="alakxender/dv-domain-lexicons",
35
- filename="dv.domain.news.small.v1.lexicon",
36
  )
37
 
38
  @spaces.GPU
 
29
  filename="language_model/5gram.bin",
30
  )
31
 
32
+ lex_files = [
33
+ "dv.domain.news.small.v1.lexicon",
34
+ "dv.domain.news.small.v2.lexicon",
35
+ "dv.domain.news.large.v1.lexicon",
36
+ "dv.domain.stories.small.v1.lexicon",
37
+ ]
38
+
39
  lexicon_file = hf_hub_download(
40
  repo_type="dataset",
41
  repo_id="alakxender/dv-domain-lexicons",
42
+ filename=lex_files[2],
43
  )
44
 
45
  @spaces.GPU