Spaces:

taesiri
/

CLIPScore

Running on Zero

taesiri commited on Sep 2

Commit

73f9f45

•

1 Parent(s): fc5e177

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -6,7 +6,7 @@ import spaces
 model = CLIPModel.from_pretrained("openai/clip-vit-base-patch16").to("cuda")
 processor = CLIPProcessor.from_pretrained("openai/clip-vit-base-patch16")
-@spaces.GPU(duration=120)
 def calculate_score(image, text):
     labels = text.split(";")
     labels = [l.strip() for l in labels]

 model = CLIPModel.from_pretrained("openai/clip-vit-base-patch16").to("cuda")
 processor = CLIPProcessor.from_pretrained("openai/clip-vit-base-patch16")
+@spaces.GPU
 def calculate_score(image, text):
     labels = text.split(";")
     labels = [l.strip() for l in labels]