Hugging Face's logo

Spaces:
taesiri
/
CLIPScore
Running on Zero

App Files Community
taesiri commited on
Commit
73f9f45
·
verified ·
1 Parent(s): fc5e177

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -6,7 +6,7 @@ import spaces
6
  model = CLIPModel.from_pretrained("openai/clip-vit-base-patch16").to("cuda")
7
  processor = CLIPProcessor.from_pretrained("openai/clip-vit-base-patch16")
8
 
9
- @spaces.GPU(duration=120)
10
  def calculate_score(image, text):
11
  labels = text.split(";")
12
  labels = [l.strip() for l in labels]
 
6
  model = CLIPModel.from_pretrained("openai/clip-vit-base-patch16").to("cuda")
7
  processor = CLIPProcessor.from_pretrained("openai/clip-vit-base-patch16")
8
 
9
+ @spaces.GPU
10
  def calculate_score(image, text):
11
  labels = text.split(";")
12
  labels = [l.strip() for l in labels]