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