Spaces:

sergiopaniego
/

sam_vs_sam_hq_image_comparison

Running on L4

sergiopaniego HF Staff commited on 15 days ago

Commit

f9f0b68

1 Parent(s): a09a58c

Update

Files changed (1) hide show

app.py CHANGED Viewed

@@ -1,6 +1,6 @@
 import gradio as gr
 import torch
-#import spaces
 import json
 import base64
 from io import BytesIO
@@ -18,7 +18,7 @@ sam_hq_processor = SamHQProcessor.from_pretrained("syscv-community/sam-hq-vit-hu
 sam_model = SamModel.from_pretrained("facebook/sam-vit-huge")
 sam_processor = SamProcessor.from_pretrained("facebook/sam-vit-huge")
-#@spaces.GPU
 def predict_masks_and_scores(model, processor, raw_image, input_points=None, input_boxes=None):
     if input_boxes is not None:
         input_boxes = [input_boxes]

 import gradio as gr
 import torch
+import spaces
 import json
 import base64
 from io import BytesIO
 sam_model = SamModel.from_pretrained("facebook/sam-vit-huge")
 sam_processor = SamProcessor.from_pretrained("facebook/sam-vit-huge")
+@spaces.GPU
 def predict_masks_and_scores(model, processor, raw_image, input_points=None, input_boxes=None):
     if input_boxes is not None:
         input_boxes = [input_boxes]