Spaces:
Sleeping
Sleeping
aman-augurs
commited on
Commit
•
98204a1
1
Parent(s):
fe0fe5b
Update app.py
Browse files
app.py
CHANGED
@@ -6,7 +6,7 @@ from transformers import AutoTokenizer, AutoModelForCausalLM
|
|
6 |
def load_model():
|
7 |
try:
|
8 |
# Define the model path (Falcon 7B with quantization applied)
|
9 |
-
base_model = "aman-augurs/
|
10 |
|
11 |
# Load the tokenizer (works for CPU)
|
12 |
tokenizer = AutoTokenizer.from_pretrained(base_model)
|
|
|
6 |
def load_model():
|
7 |
try:
|
8 |
# Define the model path (Falcon 7B with quantization applied)
|
9 |
+
base_model = "aman-augurs/llama3.2-30-fine-tuned"
|
10 |
|
11 |
# Load the tokenizer (works for CPU)
|
12 |
tokenizer = AutoTokenizer.from_pretrained(base_model)
|