Update app.py

app.py

@@ -10,7 +10,7 @@ hf_token = os.getenv("HF_API_TOKEN")
 base_model_name = "larry1129/WooWoof_AI_Vision_merged_16bit_3b"
 
 # 定义 adapter 模型名称
-# adapter_model_name = "larry1129/WooWoof_AI"
+adapter_model_name = "larry1129/WooWoof_AI_Vision_merged_16bit_3b"
 
 # 定义全局变量用于缓存模型和分词器
 model = None
@@ -77,7 +77,7 @@ def generate_response(instruction, input_text):
             # 加载适配器模型
             model = PeftModel.from_pretrained(
                 base_model,
-                 # adapter_model_name,
+                 adapter_model_name,
                 torch_dtype=torch.float16,
                 use_auth_token=hf_token
             )