Update app.py

app.py

@@ -5,14 +5,13 @@ import spaces
 
 
 # Check if CUDA is available and set the device accordingly
-device = "cuda" if torch.cuda.is_available() else "cpu"
-
-model = CLIPModel.from_pretrained("openai/clip-vit-base-patch16").to(device)
+model = CLIPModel.from_pretrained("openai/clip-vit-base-patch16").to("cuda")
 processor = CLIPProcessor.from_pretrained("openai/clip-vit-base-patch16")
 
 
-@spaces.GPU
+@spaces.GPU(duration=120)
 def calculate_score(image, text):
+    print(device)
     labels = text.split(";")
     labels = [l.strip() for l in labels]
     labels = list(filter(None, labels))