Update app.py

app.py

@@ -15,6 +15,9 @@ print(device)
 Business_Listing = "Air Guide"
 
 def find_similarity(image1, image2, text_input):
+    if image1 is None or image2 is None:
+        return None
+
     image_features = []
 
     # Preprocess and encode the two images
@@ -36,7 +39,7 @@ def find_similarity(image1, image2, text_input):
     # Calculate cosine similarity for each image
     for image_feature in image_features:
         image_feature /= image_feature.norm(dim=-1, keepdim=True)
-        similarity = (text_features @ image_feature.T).cpu().numpy()
+        similarity = (text_features @ image_feature.T).cpu().detach().numpy()
         similarities.append(similarity[0, 0])
 
     # Determine which image has a higher similarity to the text