app.py segment anything pip install

app.py

@@ -1,9 +1,16 @@
 try:
     import detectron2
-except:
+except ImportError:
     import os
     os.system('pip install git+https://github.com/facebookresearch/detectron2.git')
 
+try:
+    from segment_anything import build_sam, SamPredictor
+except ImportError:
+    import os
+    os.system('pip install git+https://github.com/facebookresearch/segment-anything.git')
+    from segment_anything import build_sam, SamPredictor
+
 import gradio as gr
 import json
 import numpy as np