aman-augurs commited on
Commit
98204a1
1 Parent(s): fe0fe5b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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/falcon7b-final-merged-quantized-fine-tuned"
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)