Update app.py
Browse files
app.py
CHANGED
@@ -31,14 +31,15 @@ st.set_page_config(
|
|
31 |
)
|
32 |
|
33 |
# Model base and options
|
34 |
-
BASE_MODEL_NAME = "
|
|
|
35 |
MODEL_OPTIONS = {
|
36 |
-
"Full Fine-Tuned": "amiguel/mistral-angolan-laborlaw-bert-base-pt", #"amiguel/mistral-angolan-laborlaw-gpt2",#"amiguel/mistral-angolan-laborlaw-ptt5", #"amiguel/mistral-angolan-laborlaw",
|
37 |
"LoRA Adapter": "amiguel/SmolLM2-360M-concise-reasoning-lora",
|
38 |
"QLoRA Adapter": "amiguel/SmolLM2-360M-concise-reasoning-qlora"
|
39 |
}
|
40 |
|
41 |
-
st.title("π Assistente
|
42 |
|
43 |
USER_AVATAR = "https://raw.githubusercontent.com/achilela/vila_fofoka_analysis/9904d9a0d445ab0488cf7395cb863cce7621d897/USER_AVATAR.png"
|
44 |
BOT_AVATAR = "https://raw.githubusercontent.com/achilela/vila_fofoka_analysis/991f4c6e4e1dc7a8e24876ca5aae5228bcdb4dba/Ataliba_Avatar.jpg"
|
|
|
31 |
)
|
32 |
|
33 |
# Model base and options
|
34 |
+
BASE_MODEL_NAME = "
|
35 |
+
"NousResearch/Nous-Hermes-2-Mistral-7B-DPO" #"neuralmind/bert-base-portuguese-cased" #"pierreguillou/gpt2-small-portuguese" #"unicamp-dl/ptt5-base-portuguese-vocab" #"mistralai/Mistral-7B-Instruct-v0.2"
|
36 |
MODEL_OPTIONS = {
|
37 |
+
"Full Fine-Tuned": "amiguel/GM_finetune", #"amiguel/mistral-angolan-laborlaw-bert-base-pt", #"amiguel/mistral-angolan-laborlaw-gpt2",#"amiguel/mistral-angolan-laborlaw-ptt5", #"amiguel/mistral-angolan-laborlaw",
|
38 |
"LoRA Adapter": "amiguel/SmolLM2-360M-concise-reasoning-lora",
|
39 |
"QLoRA Adapter": "amiguel/SmolLM2-360M-concise-reasoning-qlora"
|
40 |
}
|
41 |
|
42 |
+
st.title("π Assistente | Angola π")
|
43 |
|
44 |
USER_AVATAR = "https://raw.githubusercontent.com/achilela/vila_fofoka_analysis/9904d9a0d445ab0488cf7395cb863cce7621d897/USER_AVATAR.png"
|
45 |
BOT_AVATAR = "https://raw.githubusercontent.com/achilela/vila_fofoka_analysis/991f4c6e4e1dc7a8e24876ca5aae5228bcdb4dba/Ataliba_Avatar.jpg"
|