Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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,
|