amiguel commited on
Commit
fe620e6
Β·
verified Β·
1 Parent(s): e4023b2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -28,13 +28,14 @@ if not HF_TOKEN:
28
  raise ValueError("Missing Hugging Face Token. Please set the HF_TOKEN environment variable.")
29
 
30
  # πŸŽ› Model base and adapters
31
- BASE_MODEL_NAME = "neuralmind/bert-base-portuguese-cased"
32
  MODEL_OPTIONS = {
33
- "Full Fine-Tuned": "amiguel/mistral-angolan-laborlaw-bert-base-pt",
34
  "LoRA Adapter": "amiguel/SmolLM2-360M-concise-reasoning-lora",
35
  "QLoRA Adapter": "amiguel/SmolLM2-360M-concise-reasoning-qlora"
36
  }
37
 
 
38
  # πŸ–Ό UI Setup
39
  st.set_page_config(page_title="Assistente LGT | Angola", page_icon="πŸš€", layout="centered")
40
  st.title("πŸš€ Assistente LGT | Angola πŸš€")
 
28
  raise ValueError("Missing Hugging Face Token. Please set the HF_TOKEN environment variable.")
29
 
30
  # πŸŽ› Model base and adapters
31
+ BASE_MODEL_NAME = "unicamp-dl/ptt5-base-portuguese-vocab" #"neuralmind/bert-base-portuguese-cased" #"pierreguillou/gpt2-small-portuguese" # #"mistralai/Mistral-7B-Instruct-v0.2"
32
  MODEL_OPTIONS = {
33
+ "Full Fine-Tuned": "amiguel/mistral-angolan-laborlaw-ptt5" #"amiguel/mistral-angolan-laborlaw-bert-base-pt", #"amiguel/mistral-angolan-laborlaw-gpt2",#, #"amiguel/mistral-angolan-laborlaw",
34
  "LoRA Adapter": "amiguel/SmolLM2-360M-concise-reasoning-lora",
35
  "QLoRA Adapter": "amiguel/SmolLM2-360M-concise-reasoning-qlora"
36
  }
37
 
38
+
39
  # πŸ–Ό UI Setup
40
  st.set_page_config(page_title="Assistente LGT | Angola", page_icon="πŸš€", layout="centered")
41
  st.title("πŸš€ Assistente LGT | Angola πŸš€")