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:
|