Update app.py

app.py

@@ -10,10 +10,10 @@ HAS_GPU = False
 # Model title and context size limit
 ctx_limit = 4000
 title = "1B5"
-model_file = "RWKV-5-World-1B5-v2-20231025-ctx4096"
+#model_file = "RWKV-5-World-1B5-v2-20231025-ctx4096"
 
 #title = "RWKV-4-World"
-#model_file = "RWKV-4-World-1.5B-v1-fixed-20230612-ctx4096"
+model_file = "rwkv-4-world/RWKV-4-World-CHNtuned-1.5B-v1-20230620-ctx4096.pth"
 
 # Get the GPU count
 try:
@@ -39,8 +39,8 @@ if HAS_GPU == True :
 
 # Load the model accordingly
 from rwkv.model import RWKV
-model_path = hf_hub_download(repo_id="BlinkDL/rwkv-5-world", filename=f"{model_file}.pth")
-#model_path = hf_hub_download(repo_id="BlinkDL/rwkv-4-world", filename=f"{model_file}.pth")
+#model_path = hf_hub_download(repo_id="BlinkDL/rwkv-5-world", filename=f"{model_file}.pth")
+model_path = hf_hub_download(repo_id="BlinkDL/rwkv-4-world", filename=f"{model_file}.pth")
 model = RWKV(model=model_path, strategy=MODEL_STRAT)
 from rwkv.utils import PIPELINE, PIPELINE_ARGS
 pipeline = PIPELINE(model, "rwkv_vocab_v20230424")