Update app.py

app.py

@@ -5,14 +5,14 @@ import difflib
 from threading import Thread
 from transformers import AutoModel, AutoTokenizer, AutoModelForCausalLM, AutoModelForSeq2SeqLM, TextIteratorStreamer
 
-model_id = "textcleanlm/textcleanlm-1-4b"
+model_id = "textcleanlm/0.6b-sft"
 model = None
 tokenizer = None
 
 def load_model():
     global model, tokenizer
     if model is None:
-        tokenizer = AutoTokenizer.from_pretrained(model_id)
+        tokenizer = AutoTokenizer.from_pretrained(model_id, subfolder="checkpoint-3717")
         
         # Add padding token if needed
         if tokenizer.pad_token is None: