Update app.py

app.py

@@ -4,7 +4,7 @@ import cv2
 import gradio as gr
 from segment_anything import sam_model_registry, SamAutomaticMaskGenerator
 from PIL import Image
-from huggingface_hub import hf_hub_download  # 导入 Hugging Face Hub 下载工具
+from huggingface_hub import hf_hub_download
 
 # 下载模型文件
 chkpt_path = hf_hub_download(
@@ -12,8 +12,8 @@ chkpt_path = hf_hub_download(
     filename="checkpoints/sam_vit_b_01ec64.pth"  # 下载 sam_vit_b_01ec64.pth 模型
 )
 
-# 加载 Segment Anything 模型
-sam = sam_model_registry["vit_b"](checkpoint=chkpt_path).to("cuda")  # 注意模型类型改为 vit_b
+# 加载 Segment Anything 模型（使用 CPU）
+sam = sam_model_registry["vit_b"](checkpoint=chkpt_path).to("cpu")  # 改为 CPU
 mask_generator = SamAutomaticMaskGenerator(sam)
 
 def segment(image):