Update app.py

app.py

@@ -3,15 +3,12 @@ import gradio as gr
 from transformers import CLIPProcessor, CLIPModel
 import spaces
 
-
-# Check if CUDA is available and set the device accordingly
 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):
-    print(device)
     labels = text.split(";")
     labels = [l.strip() for l in labels]
     labels = list(filter(None, labels))
@@ -19,8 +16,8 @@ def calculate_score(image, text):
         return dict()
     inputs = processor(text=labels, images=image, return_tensors="pt", padding=True)
     inputs = {
-        k: v.to(device) for k, v in inputs.items()
-    }  # Move tensors to the appropriate device
+        k: v.to("cuda") for k, v in inputs.items()
+    }  
     outputs = model(**inputs)
     logits_per_image = (
         outputs.logits_per_image.detach().cpu().numpy()