Update app.py

app.py

@@ -1,5 +1,6 @@
 import gradio as gr
 import os
+import subprocess
 from PIL import Image, ImageChops, ImageFilter
 from ultralytics import YOLO
 from segment_anything import SamPredictor, sam_model_registry
@@ -13,6 +14,24 @@ from segment_anything import SamPredictor, sam_model_registry
 from yolo_world.models.detectors import build_detector
 from mmcv import Config
 
+# 初始化环境
+def install_requirements():
+    try:
+        # 优先安装基础依赖
+        subprocess.run(["pip", "install", "-r", "base_requirements.txt"], check=True)
+        # 安装其他依赖
+        subprocess.run(["pip", "install", "-r", "additional_requirements.txt"], check=True)
+    except subprocess.CalledProcessError as e:
+        print(f"依赖安装失败: {e}")
+        exit(1)
+
+# 检查是否需要安装依赖
+if not os.path.exists("requirements_installed.txt"):
+    install_requirements()
+    # 创建标志文件，避免重复安装
+    with open("requirements_installed.txt", "w") as f:
+        f.write("installed")
+
 # 初始化模型
 clip_model = CLIPModel.from_pretrained("openai/clip-vit-base-patch32")
 clip_processor = CLIPProcessor.from_pretrained("openai/clip-vit-base-patch32")