Hugging Face's logo

Spaces:
clarin-knext
/
dialogue-state-tracking
Runtime error

App Files Community
1
ajanz commited on
Commit
b0f59a3
·
verified ·
1 Parent(s): 46848cd

Update models.py

Browse files
Files changed (1) hide show
  1. models.py +4 -34
models.py CHANGED
@@ -22,41 +22,11 @@ DEFAULT_DST_INPUTS: Dict[str, str] = {
22
 
23
 
24
  DST_MODELS: Dict[str, Dict[str, Any]] = {
25
- # "plt5-small": {
26
- # "model": T5ForConditionalGeneration.from_pretrained("clarin-knext/plt5-small-dst", use_auth_token=auth_token),
27
- # "tokenizer": T5Tokenizer.from_pretrained("clarin-knext/plt5-small-dst", use_auth_token=auth_token),
28
- # "default_input": DEFAULT_DST_INPUTS["polish"],
29
- # },
30
- # "plt5-base": {
31
- # "model": T5ForConditionalGeneration.from_pretrained("clarin-knext/plt5-base-dst", use_auth_token=auth_token),
32
- # "tokenizer": T5Tokenizer.from_pretrained("clarin-knext/plt5-base-dst", use_auth_token=auth_token),
33
- # "default_input": DEFAULT_DST_INPUTS["polish"],
34
- # },
35
- "plt5-base-poquad-dst-v2": {
36
- "model": T5ForConditionalGeneration.from_pretrained("clarin-knext/plt5-base-poquad-dst-v2", use_auth_token=auth_token),
37
- "tokenizer": T5Tokenizer.from_pretrained("clarin-knext/plt5-base-poquad-dst-v2", use_auth_token=auth_token),
38
  "default_input": DEFAULT_DST_INPUTS["polish"],
39
- },
40
- # "t5-small": {
41
- # "model": T5ForConditionalGeneration.from_pretrained("clarin-knext/t5-small-dst", use_auth_token=auth_token),
42
- # "tokenizer": T5Tokenizer.from_pretrained("clarin-knext/t5-small-dst", use_auth_token=auth_token),
43
- # "default_input": DEFAULT_DST_INPUTS["english"],
44
- # },
45
- # "t5-base": {
46
- # "model": T5ForConditionalGeneration.from_pretrained("clarin-knext/t5-base-dst", use_auth_token=auth_token),
47
- # "tokenizer": T5Tokenizer.from_pretrained("clarin-knext/t5-base-dst", use_auth_token=auth_token),
48
- # "default_input": DEFAULT_DST_INPUTS["english"],
49
- # },
50
- # "flant5-small [EN/PL]": {
51
- # "model": T5ForConditionalGeneration.from_pretrained("clarin-knext/flant5-small-dst", use_auth_token=auth_token),
52
- # "tokenizer": T5Tokenizer.from_pretrained("clarin-knext/flant5-small-dst", use_auth_token=auth_token),
53
- # "default_input": DEFAULT_DST_INPUTS["english"],
54
- # },
55
- # "flant5-base [EN/PL]": {
56
- # "model": T5ForConditionalGeneration.from_pretrained("clarin-knext/flant5-base-dst", use_auth_token=auth_token),
57
- # "tokenizer": T5Tokenizer.from_pretrained("clarin-knext/flant5-base-dst", use_auth_token=auth_token),
58
- # "default_input": DEFAULT_DST_INPUTS["english"],
59
- # },
60
  }
61
 
62
 
 
22
 
23
 
24
  DST_MODELS: Dict[str, Dict[str, Any]] = {
25
+ "plt5-large-poquad-dst-v2": {
26
+ "model": T5ForConditionalGeneration.from_pretrained("clarin-knext/plt5-large-poquad-dst-v2", use_auth_token=auth_token),
27
+ "tokenizer": T5Tokenizer.from_pretrained("clarin-knext/plt5-large-poquad-dst-v2", use_auth_token=auth_token),
 
 
 
 
 
 
 
 
 
 
28
  "default_input": DEFAULT_DST_INPUTS["polish"],
29
+ }
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
30
  }
31
 
32