Spaces:

beingcognitive
/

segmentation_w_object_detection

Runtime error

beingcognitive commited on Mar 23, 2024

Commit

57aedf3

verified ·

1 Parent(s): 9785b43

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -10,7 +10,8 @@ import requests
 from io import BytesIO
 def main():
-    st.title("Image Segmentation")
     # Load SAM by Facebook
     processor = AutoProcessor.from_pretrained("facebook/sam-vit-huge")

 from io import BytesIO
 def main():
+    st.title("Image Segmentation w/ Object Detection")
+    st.subheader("models : facebook/sam-vit-huge, facebook/detr-resnet-50")
     # Load SAM by Facebook
     processor = AutoProcessor.from_pretrained("facebook/sam-vit-huge")