Update app.py

app.py

@@ -14,13 +14,12 @@ from llava.model.builder import load_pretrained_model
 from llava.utils import disable_torch_init
 from llava.mm_utils import tokenizer_image_token
 
-# 确保 Hugging Face 缓存目录设置正确
+# **确保 Hugging Face 缓存目录正确**
 os.environ["HUGGINGFACE_HUB_CACHE"] = os.getcwd() + "/weights"
 
 # **加载 LLaVA-1.5-13B**
 disable_torch_init()
-model_id = "Yanqing0327/LLaVA-project"
-
+model_id = "Yanqing0327/LLaVA-project"  # 替换为你的 Hugging Face 模型仓库
 tokenizer, model, image_processor, context_len = load_pretrained_model(
     model_id, model_name="llava-v1.5-13b", model_base=None, load_8bit=False, load_4bit=False
 )
@@ -28,7 +27,6 @@ tokenizer, model, image_processor, context_len = load_pretrained_model(
 device = "cuda" if torch.cuda.is_available() else "cpu"
 model = model.to(device)
 
-
 def load_image(image_file):
     """加载本地图片或 URL 图片"""
     if isinstance(image_file, str) and (image_file.startswith('http') or image_file.startswith('https')):
@@ -38,7 +36,6 @@ def load_image(image_file):
         image = Image.open(image_file).convert('RGB')
     return image
 
-
 def llava_infer(image, text, temperature, top_p, max_tokens):
     """LLaVA 模型推理"""
     if image is None or text.strip() == "":