amaltese commited on
Commit
590d167
·
verified ·
1 Parent(s): f3b5025

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -105,7 +105,7 @@ def finetune_model(csv_file, sample_size=100, num_epochs=3, progress=gr.Progress
105
  progress(0.2, "Loading tokenizer...")
106
 
107
  # Try the non-gated Mistral model first
108
- model_name = "mistralai/Mistral-7B-Instruct-v0.2"
109
  try:
110
  tokenizer = AutoTokenizer.from_pretrained(model_name, token=hf_token)
111
  print(f"Successfully loaded tokenizer for {model_name}")
 
105
  progress(0.2, "Loading tokenizer...")
106
 
107
  # Try the non-gated Mistral model first
108
+ model_name = "mistralai/Mistral-7B-v0.1"
109
  try:
110
  tokenizer = AutoTokenizer.from_pretrained(model_name, token=hf_token)
111
  print(f"Successfully loaded tokenizer for {model_name}")