anvilarth commited on
Commit
a486839
·
verified ·
1 Parent(s): 35e1642

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -251,7 +251,7 @@ class GradioWindow():
251
  self.concatenated_masks = res
252
  return res, current_object, True
253
 
254
-
255
  def detect(self, image: Image, prompt: str, is_segmmask: bool,
256
  box_threshold: float, text_threshold: float):
257
  detections = self.grounding_dino_model.predict_with_classes(
 
251
  self.concatenated_masks = res
252
  return res, current_object, True
253
 
254
+ @spaces.GPU
255
  def detect(self, image: Image, prompt: str, is_segmmask: bool,
256
  box_threshold: float, text_threshold: float):
257
  detections = self.grounding_dino_model.predict_with_classes(