Spaces:
Running
on
L4
Running
on
L4
Commit
·
ca815e1
1
Parent(s):
d6ff06e
Updated
Browse files
app.py
CHANGED
@@ -1,6 +1,6 @@
|
|
1 |
import gradio as gr
|
2 |
import torch
|
3 |
-
import spaces
|
4 |
import json
|
5 |
import base64
|
6 |
from io import BytesIO
|
@@ -17,7 +17,7 @@ sam_hq_processor = SamHQProcessor.from_pretrained("syscv-community/sam-hq-vit-hu
|
|
17 |
sam_model = SamModel.from_pretrained("facebook/sam-vit-huge")
|
18 |
sam_processor = SamProcessor.from_pretrained("facebook/sam-vit-huge")
|
19 |
|
20 |
-
|
21 |
def predict_masks_and_scores(model, processor, raw_image, input_points=None, input_boxes=None):
|
22 |
if input_boxes is not None:
|
23 |
input_boxes = [input_boxes]
|
|
|
1 |
import gradio as gr
|
2 |
import torch
|
3 |
+
#import spaces
|
4 |
import json
|
5 |
import base64
|
6 |
from io import BytesIO
|
|
|
17 |
sam_model = SamModel.from_pretrained("facebook/sam-vit-huge")
|
18 |
sam_processor = SamProcessor.from_pretrained("facebook/sam-vit-huge")
|
19 |
|
20 |
+
#@spaces.GPU
|
21 |
def predict_masks_and_scores(model, processor, raw_image, input_points=None, input_boxes=None):
|
22 |
if input_boxes is not None:
|
23 |
input_boxes = [input_boxes]
|
utils.py
CHANGED
@@ -132,7 +132,8 @@ def show_all_annotations_on_image_base64(raw_image, masks=None, scores=None, box
|
|
132 |
score = scores[0].item()
|
133 |
else:
|
134 |
score = float(np.array(scores).flatten()[0])
|
135 |
-
ax.set_title(f"{model_name} - Score: {score:.3f}")
|
|
|
136 |
|
137 |
|
138 |
if input_points is not None:
|
|
|
132 |
score = scores[0].item()
|
133 |
else:
|
134 |
score = float(np.array(scores).flatten()[0])
|
135 |
+
#ax.set_title(f"{model_name} - Score: {score:.3f}")
|
136 |
+
ax.set_title(f"{model_name}")
|
137 |
|
138 |
|
139 |
if input_points is not None:
|