Hugging Face's logo

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

App Files Community
1
michalilski commited on
Commit
857aac4
•
1 Parent(s): 9388f53

plt5-base-poquad

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