amiguel commited on
Commit
b9ea175
Β·
verified Β·
1 Parent(s): 699a812

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -3
app.py CHANGED
@@ -31,14 +31,15 @@ st.set_page_config(
31
  )
32
 
33
  # Model base and options
34
- BASE_MODEL_NAME = "neuralmind/bert-base-portuguese-cased" #"pierreguillou/gpt2-small-portuguese" #"unicamp-dl/ptt5-base-portuguese-vocab" #"mistralai/Mistral-7B-Instruct-v0.2"
 
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 LGT | Angola πŸš€")
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"