Update app.py

app.py

@@ -3,26 +3,21 @@ from infer.worldmodel import Worldinfer
 from PIL import Image
 from huggingface_hub import hf_hub_download, snapshot_download
 import re
+import html 
 # 初始化模型
-llm_path = hf_hub_download(repo_id="WorldRWKV/RWKV7-0.4B-siglip2", repo_type="model", filename="rwkv-0.pth",local_dir="./")
-llm_path = "./rwkv-0"
+llm_path = hf_hub_download(repo_id="WorldRWKV/RWKV7-0.4B-G1-SigLIP2-ColdStart",filename="rwkv-0.pth",local_dir="./model_weights/")
 encoder_path = snapshot_download(repo_id="google/siglip2-base-patch16-384")
-
+# llm_path = "/mnt/B8E84E9EE84E5B30/rwkv-models/world_rwkv/world_weights/rwkv-0"
+# encoder_path = "/mnt/B8E84E9EE84E5B30/rwkv-models/world_rwkv/siglip2-base-patch16-384/"
 encoder_type = 'siglip'
 
 # 全局变量存储当前上传的图片和模型状态
 current_image = None
 current_state = None 
 first_question = False # 存储模型状态
-# 是否是第一轮对话
 # 初始化模型
 model = Worldinfer(model_path=llm_path, encoder_type=encoder_type, encoder_path=encoder_path)
 
-# 处理用户输入的核心逻辑
-import html  # 导入html库
-
-import re
-
 # 处理用户输入的核心逻辑
 def chat_fn(user_input, chat_history, image=None):
     global current_image, current_state, first_question
@@ -197,4 +192,4 @@ with gr.Blocks(title="WORLD RWKV", theme=gr.themes.Soft()) as demo:
 
 # 启动应用
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch(server_name="127.0.0.1", server_port=7860)