wondervictor commited on
Commit
47db2fb
·
verified ·
1 Parent(s): 367ad6d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -10,7 +10,7 @@ import os.path as osp
10
  from io import BytesIO
11
  from functools import partial
12
 
13
- import spaces
14
 
15
  from mmengine.runner import Runner
16
  from mmengine.dataset import Compose
@@ -53,7 +53,7 @@ class LabelAnnotator(sv.LabelAnnotator):
53
  LABEL_ANNOTATOR = LabelAnnotator(text_padding=4,
54
  text_scale=0.5,
55
  text_thickness=1)
56
- @spaces.GPU
57
  def generate_image_embeddings(prompt_image,
58
  vision_encoder,
59
  vision_processor,
@@ -71,7 +71,7 @@ def generate_image_embeddings(prompt_image,
71
  img_feats = projector(img_feats)
72
  return img_feats
73
 
74
- @spaces.GPU
75
  def run_image(runner,
76
  vision_encoder,
77
  vision_processor,
 
10
  from io import BytesIO
11
  from functools import partial
12
 
13
+ # import spaces
14
 
15
  from mmengine.runner import Runner
16
  from mmengine.dataset import Compose
 
53
  LABEL_ANNOTATOR = LabelAnnotator(text_padding=4,
54
  text_scale=0.5,
55
  text_thickness=1)
56
+ # @spaces.GPU
57
  def generate_image_embeddings(prompt_image,
58
  vision_encoder,
59
  vision_processor,
 
71
  img_feats = projector(img_feats)
72
  return img_feats
73
 
74
+ # @spaces.GPU
75
  def run_image(runner,
76
  vision_encoder,
77
  vision_processor,