amiguel commited on
Commit
11a54d6
·
verified ·
1 Parent(s): 9e79c7a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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
  }