TrabbyPatty commited on
Commit
2355649
Β·
verified Β·
1 Parent(s): c5552a7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -1
app.py CHANGED
@@ -8,7 +8,12 @@ model_id = "TrabbyPatty/mistral-7b-instruct-finetuned-flashcards"
8
  hf_token = os.getenv("alluse") # Hugging Face token from Space secrets
9
 
10
  # === Load tokenizer & model with authentication ===
11
- tokenizer = AutoTokenizer.from_pretrained(model_id, token=hf_token)
 
 
 
 
 
12
  model = AutoModelForCausalLM.from_pretrained(
13
  model_id,
14
  device_map="cpu", # force CPU
 
8
  hf_token = os.getenv("alluse") # Hugging Face token from Space secrets
9
 
10
  # === Load tokenizer & model with authentication ===
11
+ tokenizer = AutoTokenizer.from_pretrained(
12
+ model_id,
13
+ token=hf_token,
14
+ use_fast=False # force slow tokenizer (fixes error)
15
+ )
16
+
17
  model = AutoModelForCausalLM.from_pretrained(
18
  model_id,
19
  device_map="cpu", # force CPU