Update app.py

app.py

@@ -20,6 +20,17 @@ MODELS = {
     "0.4B (Larger)": "RWKV-x070-World-0.4B-v2.9-20250107-ctx4096.pth"
 }
 
+# Download tokenizer if not present
+TOKENIZER_FILE = "rwkv_vocab_v20230424.txt"
+TOKENIZER_URL = "https://raw.githubusercontent.com/BlinkDL/ChatRWKV/main/v2/rwkv_vocab_v20230424.txt"
+
+def download_tokenizer():
+    if not os.path.exists(TOKENIZER_FILE):
+        print("Downloading tokenizer...")
+        response = requests.get(TOKENIZER_URL)
+        with open(TOKENIZER_FILE, 'wb') as f:
+            f.write(response.content)
+
 def download_model(model_name):
     """Download model if not present"""
     if not os.path.exists(model_name):
@@ -48,11 +59,16 @@ class ModelManager:
     def load_model(self, model_name):
         if model_name != self.current_model_name:
             download_model(MODELS[model_name])
-            self.current_model = RWKV(model=MODELS[model_name], strategy='cpu fp32')
-            self.pipeline = PIPELINE(self.current_model, "rwkv_vocab_v20230424")
+            self.current_model = RWKV(
+                model=MODELS[model_name],
+                strategy='cpu fp32',
+                v7=True
+            )
+            self.pipeline = PIPELINE(self.current_model, TOKENIZER_FILE)
             self.current_model_name = model_name
         return self.pipeline
 
+
 model_manager = ModelManager()
 
 def generate_response(