Update app.py

app.py

@@ -10,7 +10,7 @@ HAS_GPU = False
 # Model title and context size limit
 ctx_limit = 2000
 title = "RWKV-5-World-1B5-v2-20231025-ctx4096"
-model_file = "rwkv-5-h-world-1b5-60%-test"
+model_file = "rwkv-5-h-world-1b5"
 
 # Get the GPU count
 try:
@@ -36,7 +36,7 @@ if HAS_GPU == True :
 
 # Load the model accordingly
 from rwkv.model import RWKV
-model_path = hf_hub_download(repo_id="a686d380/rwkv-h-world-1b5-test", filename=f"{model_file}.pth")
+model_path = hf_hub_download(repo_id="a686d380/rwkv-5-h-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")