0.6b-demo

Running on Zero

sumuks commited on 9 days ago

Commit

443499e

verified ·

1 Parent(s): fbccfd2

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -12,7 +12,7 @@ tokenizer = None
 def load_model():
     global model, tokenizer
     if model is None:
-        tokenizer = AutoTokenizer.from_pretrained(model_id, subfolder="checkpoint-3717")
         # Add padding token if needed
         if tokenizer.pad_token is None:

 def load_model():
     global model, tokenizer
     if model is None:
+        tokenizer = AutoTokenizer.from_pretrained(model_id)
         # Add padding token if needed
         if tokenizer.pad_token is None: