5m4ck3r commited on
Commit
e7de587
·
verified ·
1 Parent(s): c50f5c0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -7,6 +7,7 @@ import base64
7
  from PIL import Image, ImageDraw
8
  from io import BytesIO
9
  from sentence_transformers import SentenceTransformer, util
 
10
 
11
  backgroundPipe = pipeline("image-segmentation", model="facebook/maskformer-swin-large-coco")
12
  PersonPipe = pipeline("image-segmentation", model="mattmdjaga/segformer_b2_clothes")
@@ -95,6 +96,7 @@ def process_image(image):
95
  processed_image.putdata(final_data)
96
  return processed_image
97
 
 
98
  def processAndGetMask(image: str, text: str):
99
  datas = getImageDetails(image)
100
  labs = list(datas.keys())
 
7
  from PIL import Image, ImageDraw
8
  from io import BytesIO
9
  from sentence_transformers import SentenceTransformer, util
10
+ import spaces
11
 
12
  backgroundPipe = pipeline("image-segmentation", model="facebook/maskformer-swin-large-coco")
13
  PersonPipe = pipeline("image-segmentation", model="mattmdjaga/segformer_b2_clothes")
 
96
  processed_image.putdata(final_data)
97
  return processed_image
98
 
99
+ @spaces.GPU()
100
  def processAndGetMask(image: str, text: str):
101
  datas = getImageDetails(image)
102
  labs = list(datas.keys())