Spaces:

sergiopaniego
/

sam_vs_sam_hq_image_comparison

Running on L4

sergiopaniego HF Staff commited on 18 days ago

Commit

941f0af

verified ·

1 Parent(s): ec9acd8

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -17,9 +17,9 @@ from PIL import Image
 from gradio_image_prompter import ImagePrompter
-sam_hq_model = SamHQModel.from_pretrained("syscv-community/sam-hq-vit-base", device_map="cuda", torch_dtype="auto")
 sam_hq_processor = SamHQProcessor.from_pretrained("syscv-community/sam-hq-vit-base")
-sam_model = SamModel.from_pretrained("facebook/sam-vit-base", device_map="cuda", torch_dtype="auto")
 sam_processor = SamProcessor.from_pretrained("facebook/sam-vit-base")
 @spaces.GPU

 from gradio_image_prompter import ImagePrompter
+sam_hq_model = SamHQModel.from_pretrained("syscv-community/sam-hq-vit-base", device_map="auto", torch_dtype="auto")
 sam_hq_processor = SamHQProcessor.from_pretrained("syscv-community/sam-hq-vit-base")
+sam_model = SamModel.from_pretrained("facebook/sam-vit-base", device_map="auto", torch_dtype="auto")
 sam_processor = SamProcessor.from_pretrained("facebook/sam-vit-base")
 @spaces.GPU