fix for running on cpu

app.py

@@ -24,17 +24,18 @@ def calculate_similarity(image_features, text_feature, lambda_val=0.5):
 
 # Load precomputed features
 features = load_features('features/patternnet_clip.pkl')
-image_features = torch.tensor(features['feats']).cuda()
+image_features = torch.tensor(features['feats'])#.cuda()
 image_paths = features['paths']
 
 def image_text_retrieval(image, text, lambda_val):
     # Preprocess image
     preprocess = open_clip.get_preprocess('ViT-L-14')
-    image = preprocess(image).unsqueeze(0).cuda()
+    image = preprocess(image).unsqueeze(0)#.cuda()
 
     # Encode image and text
     image_feature = model.encode_image(image).cpu()
-    text_feature = model.encode_text(tokenizer(text).unsqueeze(0).cuda()).cpu()
+    #text_feature = model.encode_text(tokenizer(text).unsqueeze(0).cuda()).cpu()
+    text_feature = model.encode_text(tokenizer(text).unsqueeze(0)).cpu()
 
     # Calculate combined similarities
     similarities = calculate_similarity(image_features, text_feature, lambda_val)