Hugging Face's logo

Spaces:
sergiopaniego
/
sam_vs_sam_hq_image_comparison
Running on L4

App Files Community
sergiopaniego HF Staff commited on
Commit
2569c33
·
1 Parent(s): 296f4e8

Updated models

Browse files
Files changed (1) hide show
  1. app.py +9 -5
app.py CHANGED
@@ -16,11 +16,15 @@ from utils import *
16
  from PIL import Image
17
  from gradio_image_prompter import ImagePrompter
18
 
19
- sam_hq_model = SamHQModel.from_pretrained("syscv-community/sam-hq-vit-huge")
20
- sam_hq_processor = SamHQProcessor.from_pretrained("syscv-community/sam-hq-vit-huge")
21
-
22
- sam_model = SamModel.from_pretrained("facebook/sam-vit-huge")
23
- sam_processor = SamProcessor.from_pretrained("facebook/sam-vit-huge")
 
 
 
 
24
 
25
  @spaces.GPU
26
  def predict_masks_and_scores(model, processor, raw_image, input_points=None, input_boxes=None):
 
16
  from PIL import Image
17
  from gradio_image_prompter import ImagePrompter
18
 
19
+ #sam_hq_model = SamHQModel.from_pretrained("syscv-community/sam-hq-vit-huge")
20
+ #sam_hq_processor = SamHQProcessor.from_pretrained("syscv-community/sam-hq-vit-huge")
21
+ sam_hq_model = SamHQModel.from_pretrained("syscv-community/sam-hq-vit-base")
22
+ sam_hq_processor = SamHQProcessor.from_pretrained("syscv-community/sam-hq-vit-base")
23
+
24
+ #sam_model = SamModel.from_pretrained("facebook/sam-vit-huge")
25
+ #sam_processor = SamProcessor.from_pretrained("facebook/sam-vit-huge")
26
+ sam_model = SamModel.from_pretrained("facebook/sam-vit-base")
27
+ sam_processor = SamProcessor.from_pretrained("facebook/sam-vit-base")
28
 
29
  @spaces.GPU
30
  def predict_masks_and_scores(model, processor, raw_image, input_points=None, input_boxes=None):