Spaces:
Sleeping
Sleeping
Bram van Es
commited on
Commit
·
797ad44
1
Parent(s):
5d8e89e
added models
Browse files
config.py
CHANGED
|
@@ -16,8 +16,10 @@ DEFAULT_MODELS = {
|
|
| 16 |
"roberta-base",
|
| 17 |
"albert-base-v2"
|
| 18 |
"UMCU/CardioMedRoBERTa.nl",
|
| 19 |
-
"UMCU/
|
| 20 |
"UMCU/CardioBERTa.nl_clinical",
|
|
|
|
|
|
|
| 21 |
"CLTL/MedRoBERTa.nl",
|
| 22 |
"DTAI-KULeuven/robbert-2023-dutch-base",
|
| 23 |
"DTAI-KULeuven/robbert-2023-dutch-large"
|
|
@@ -53,8 +55,6 @@ VIZ_SETTINGS = {
|
|
| 53 |
|
| 54 |
# Processing settings
|
| 55 |
PROCESSING_SETTINGS = {
|
| 56 |
-
"default_iterations": 1,
|
| 57 |
-
"max_iterations": 10,
|
| 58 |
"epsilon": 1e-10 # Small value to avoid log(0)
|
| 59 |
}
|
| 60 |
|
|
@@ -73,32 +73,27 @@ UI_SETTINGS = {
|
|
| 73 |
{
|
| 74 |
"text": "The quick brown fox jumps over the lazy dog.",
|
| 75 |
"model": "gpt2",
|
| 76 |
-
"type": "decoder"
|
| 77 |
-
"iterations": 1
|
| 78 |
},
|
| 79 |
{
|
| 80 |
"text": "The capital of France is Paris.",
|
| 81 |
"model": "bert-base-uncased",
|
| 82 |
-
"type": "encoder"
|
| 83 |
-
"iterations": 1
|
| 84 |
},
|
| 85 |
{
|
| 86 |
"text": "Quantum entanglement defies classical physics intuition completely.",
|
| 87 |
"model": "distilgpt2",
|
| 88 |
-
"type": "decoder"
|
| 89 |
-
"iterations": 1
|
| 90 |
},
|
| 91 |
{
|
| 92 |
"text": "Buffalo buffalo Buffalo buffalo buffalo buffalo Buffalo buffalo.",
|
| 93 |
"model": "gpt2",
|
| 94 |
-
"type": "decoder"
|
| 95 |
-
"iterations": 1
|
| 96 |
},
|
| 97 |
{
|
| 98 |
"text": "Machine learning algorithms require computational resources.",
|
| 99 |
"model": "distilbert-base-uncased",
|
| 100 |
-
"type": "encoder"
|
| 101 |
-
"iterations": 1
|
| 102 |
}
|
| 103 |
]
|
| 104 |
}
|
|
|
|
| 16 |
"roberta-base",
|
| 17 |
"albert-base-v2"
|
| 18 |
"UMCU/CardioMedRoBERTa.nl",
|
| 19 |
+
"UMCU/CardioBERTa_base.nl",
|
| 20 |
"UMCU/CardioBERTa.nl_clinical",
|
| 21 |
+
"UMCU/CardioDeBERTa.nl",
|
| 22 |
+
"UMCU/CardioDeBERTa.nl_clinical",
|
| 23 |
"CLTL/MedRoBERTa.nl",
|
| 24 |
"DTAI-KULeuven/robbert-2023-dutch-base",
|
| 25 |
"DTAI-KULeuven/robbert-2023-dutch-large"
|
|
|
|
| 55 |
|
| 56 |
# Processing settings
|
| 57 |
PROCESSING_SETTINGS = {
|
|
|
|
|
|
|
| 58 |
"epsilon": 1e-10 # Small value to avoid log(0)
|
| 59 |
}
|
| 60 |
|
|
|
|
| 73 |
{
|
| 74 |
"text": "The quick brown fox jumps over the lazy dog.",
|
| 75 |
"model": "gpt2",
|
| 76 |
+
"type": "decoder"
|
|
|
|
| 77 |
},
|
| 78 |
{
|
| 79 |
"text": "The capital of France is Paris.",
|
| 80 |
"model": "bert-base-uncased",
|
| 81 |
+
"type": "encoder"
|
|
|
|
| 82 |
},
|
| 83 |
{
|
| 84 |
"text": "Quantum entanglement defies classical physics intuition completely.",
|
| 85 |
"model": "distilgpt2",
|
| 86 |
+
"type": "decoder"
|
|
|
|
| 87 |
},
|
| 88 |
{
|
| 89 |
"text": "Buffalo buffalo Buffalo buffalo buffalo buffalo Buffalo buffalo.",
|
| 90 |
"model": "gpt2",
|
| 91 |
+
"type": "decoder"
|
|
|
|
| 92 |
},
|
| 93 |
{
|
| 94 |
"text": "Machine learning algorithms require computational resources.",
|
| 95 |
"model": "distilbert-base-uncased",
|
| 96 |
+
"type": "encoder"
|
|
|
|
| 97 |
}
|
| 98 |
]
|
| 99 |
}
|