Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -32,7 +32,7 @@ if HF_TOKEN:
|
|
| 32 |
|
| 33 |
# === Available Models for Selection ===
|
| 34 |
model_map = {
|
| 35 |
-
"FinGPT LoRA"
|
| 36 |
"InvestLM (AWQ)": {"id": "yixuantt/InvestLM-mistral-AWQ", "local": False},
|
| 37 |
"FinLLaMA (LLaMA3.1-8B)": {"id": "us4/fin-llama3.1-8b", "local": False},
|
| 38 |
"FinanceConnect (13B)": {"id": "ceadar-ie/FinanceConnect-13B", "local": True},
|
|
|
|
| 32 |
|
| 33 |
# === Available Models for Selection ===
|
| 34 |
model_map = {
|
| 35 |
+
"FinGPT LoRA" : {"id": "FinGPT/fingpt-mt_llama2-7b_lora", "local": True, "custom_loader": load_fingpt_lora},
|
| 36 |
"InvestLM (AWQ)": {"id": "yixuantt/InvestLM-mistral-AWQ", "local": False},
|
| 37 |
"FinLLaMA (LLaMA3.1-8B)": {"id": "us4/fin-llama3.1-8b", "local": False},
|
| 38 |
"FinanceConnect (13B)": {"id": "ceadar-ie/FinanceConnect-13B", "local": True},
|