Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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 π")
|