Try enabling GPU if available

handler.py

@@ -5,10 +5,12 @@ from PIL import Image
 import torch
 from transformers import CLIPProcessor, CLIPTextModel, CLIPVisionModelWithProjection
 
+device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
+
 class EndpointHandler():
     def __init__(self, path=""):
-        self.text_model = CLIPTextModel.from_pretrained("rbanfield/clip-vit-large-patch14")
-        self.image_model = CLIPVisionModelWithProjection.from_pretrained("rbanfield/clip-vit-large-patch14")
+        self.text_model = CLIPTextModel.from_pretrained("rbanfield/clip-vit-large-patch14").to(device)
+        self.image_model = CLIPVisionModelWithProjection.from_pretrained("rbanfield/clip-vit-large-patch14").to(device)
         self.processor = CLIPProcessor.from_pretrained("rbanfield/clip-vit-large-patch14")
 
     def __call__(self, data):