Upload app.py with huggingface_hub
Browse files
app.py
CHANGED
|
@@ -4,7 +4,7 @@ from transformers import AutoTokenizer, AutoModelForCausalLM
|
|
| 4 |
from peft import PeftModel
|
| 5 |
import torch
|
| 6 |
|
| 7 |
-
BASE_MODEL = "
|
| 8 |
ADAPTER = "Parth03034/finbot-lora-adapter"
|
| 9 |
|
| 10 |
print("Loading tokenizer...")
|
|
@@ -54,7 +54,7 @@ def ask_finbot(message, history):
|
|
| 54 |
demo = gr.ChatInterface(
|
| 55 |
fn = ask_finbot,
|
| 56 |
title = "🤖 FinBot — Indian Finance Assistant",
|
| 57 |
-
description = "SIP, CIBIL, Nifty, Tax —
|
| 58 |
examples = ["Tell me about SIP?", "How to improve CIBIL Score?", "Explain NIFTY 50?"],
|
| 59 |
)
|
| 60 |
|
|
|
|
| 4 |
from peft import PeftModel
|
| 5 |
import torch
|
| 6 |
|
| 7 |
+
BASE_MODEL = "TinyLlama/TinyLlama-1.1B-Chat-v1.0" ###TinyLlama/TinyLlama-1.1B-Chat-v1.0 #unsloth/Llama-3.2-3B-Instruct-bnb-4bit
|
| 8 |
ADAPTER = "Parth03034/finbot-lora-adapter"
|
| 9 |
|
| 10 |
print("Loading tokenizer...")
|
|
|
|
| 54 |
demo = gr.ChatInterface(
|
| 55 |
fn = ask_finbot,
|
| 56 |
title = "🤖 FinBot — Indian Finance Assistant",
|
| 57 |
+
description = "SIP, CIBIL, Nifty, Tax — I am here to help you!",
|
| 58 |
examples = ["Tell me about SIP?", "How to improve CIBIL Score?", "Explain NIFTY 50?"],
|
| 59 |
)
|
| 60 |
|