Update app.py

app.py

@@ -8,13 +8,16 @@ import torch
 import matplotlib.pyplot as plt
 import numpy as np
 from openai import OpenAI
+from huggingface_hub import hf_hub_download
 
 # 初始化模型
 clip_model = CLIPModel.from_pretrained("openai/clip-vit-base-patch32")
 clip_processor = CLIPProcessor.from_pretrained("openai/clip-vit-base-patch32")
 blip_processor = BlipProcessor.from_pretrained("Salesforce/blip-image-captioning-base")
 blip_model = BlipForConditionalGeneration.from_pretrained("Salesforce/blip-image-captioning-base")
-sam_checkpoint = "sam_vit_h_4b8939.pth"  # 替换为实际权重路径
+sam_checkpoint = hf_hub_download(
+    repo_id="facebook/sam", filename="sam_vit_h_4b8939.pth", use_auth_token=False
+)
 sam = sam_model_registry["vit_h"](checkpoint=sam_checkpoint)
 sam_predictor = SamPredictor(sam)
 yolo_model = YOLO("yolov8x.pt")  # 替换为实际 YOLO 模型路径