sergiopaniego HF Staff commited on
Commit
ca815e1
·
1 Parent(s): d6ff06e
Files changed (2) hide show
  1. app.py +2 -2
  2. utils.py +2 -1
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
- @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]
 
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: