Update app.py

app.py

@@ -3,13 +3,11 @@ import gradio as gr
 from transformers import CLIPProcessor, CLIPModel
 import spaces
 
-model = CLIPModel.from_pretrained("openai/clip-vit-base-patch16")
+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):
-    model.to("cuda")  # Move model to CUDA inside the GPU-decorated function
-    
     labels = text.split(";")
     labels = [l.strip() for l in labels]
     labels = list(filter(None, labels))