Update app.py
Browse files
app.py
CHANGED
@@ -29,9 +29,9 @@ st.set_page_config(
|
|
29 |
)
|
30 |
|
31 |
# Model base and options
|
32 |
-
BASE_MODEL_NAME = "pierreguillou/gpt2-small-portuguese" #"unicamp-dl/ptt5-base-portuguese-vocab" #"mistralai/Mistral-7B-Instruct-v0.2"
|
33 |
MODEL_OPTIONS = {
|
34 |
-
"Full Fine-Tuned": "amiguel/mistral-angolan-laborlaw-gpt2",#"amiguel/mistral-angolan-laborlaw-ptt5", #"amiguel/mistral-angolan-laborlaw",
|
35 |
"LoRA Adapter": "amiguel/SmolLM2-360M-concise-reasoning-lora",
|
36 |
"QLoRA Adapter": "amiguel/SmolLM2-360M-concise-reasoning-qlora"
|
37 |
}
|
|
|
29 |
)
|
30 |
|
31 |
# Model base and options
|
32 |
+
BASE_MODEL_NAME = "neuralmind/bert-base-portuguese-cased" #"pierreguillou/gpt2-small-portuguese" #"unicamp-dl/ptt5-base-portuguese-vocab" #"mistralai/Mistral-7B-Instruct-v0.2"
|
33 |
MODEL_OPTIONS = {
|
34 |
+
"Full Fine-Tuned": "amiguel/mistral-angolan-laborlaw-bert-base-pt", #"amiguel/mistral-angolan-laborlaw-gpt2",#"amiguel/mistral-angolan-laborlaw-ptt5", #"amiguel/mistral-angolan-laborlaw",
|
35 |
"LoRA Adapter": "amiguel/SmolLM2-360M-concise-reasoning-lora",
|
36 |
"QLoRA Adapter": "amiguel/SmolLM2-360M-concise-reasoning-qlora"
|
37 |
}
|