merve HF staff commited on
Commit
def1b42
1 Parent(s): d86dfd1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -35,7 +35,7 @@ def extract_features_siglip(image):
35
  image_features = model.get_image_features(**inputs)
36
  return image_features
37
 
38
-
39
  def infer(input_image):
40
  input_features = extract_features_siglip(input_image["composite"].convert("RGB"))
41
  input_features = input_features.detach().cpu().numpy()
 
35
  image_features = model.get_image_features(**inputs)
36
  return image_features
37
 
38
+ @spaces.GPU
39
  def infer(input_image):
40
  input_features = extract_features_siglip(input_image["composite"].convert("RGB"))
41
  input_features = input_features.detach().cpu().numpy()