Update app.py
Browse files
app.py
CHANGED
|
@@ -9,6 +9,8 @@ import torch
|
|
| 9 |
#constants
|
| 10 |
|
| 11 |
MODEL_REGISTRY = {
|
|
|
|
|
|
|
| 12 |
"CeLLaTe3.2_Base_with_vague_weighted": "OTAR3088/CeLLaTe3.2_Base_with_vague_adapted_pubmed",
|
| 13 |
"CeLLaTe3.1_Base_no_vague_weighted": "OTAR3088/CeLLaTe3.1_Base_no_vague_adapted_pubmed_weighted",
|
| 14 |
"CeLLaTe3.1_Base_no_vague_gaz": "OTAR3088/CeLLaTe3.1_no_vague_adapted_biomed_gaz",
|
|
|
|
| 9 |
#constants
|
| 10 |
|
| 11 |
MODEL_REGISTRY = {
|
| 12 |
+
"CeLLaTe3.0_Base_with_vague_new": "OTAR3088/CeLLaTe3.0_Base_with_vague_adapted_pubmed",
|
| 13 |
+
"CeLLaTe3.0_Base_no_vague_new": "OTAR3088/CeLLaTe3.0_Base_no_vague_adapted_pubmed",
|
| 14 |
"CeLLaTe3.2_Base_with_vague_weighted": "OTAR3088/CeLLaTe3.2_Base_with_vague_adapted_pubmed",
|
| 15 |
"CeLLaTe3.1_Base_no_vague_weighted": "OTAR3088/CeLLaTe3.1_Base_no_vague_adapted_pubmed_weighted",
|
| 16 |
"CeLLaTe3.1_Base_no_vague_gaz": "OTAR3088/CeLLaTe3.1_no_vague_adapted_biomed_gaz",
|