Update app.py

app.py

@@ -217,34 +217,41 @@ def process_best_match(best_match, language):
     return best_image_url, info_html, audio_file
 
 # Function to match the uploaded image with the DataFrame to retrive the image of painting from the Datafram and it story in text and audio
+
 def compare_images(image, language):
     try:
+
         inputs = processor(images=image, return_tensors="pt")
         inputs = {k: v.to(device) for k, v in inputs.items()}
         image_features = model.get_image_features(**inputs).to(device)
 
-        best_score = -2.0
-        best_match_idx = None
+        best_score = -2.0 
+        best_match_idx = None  
 
+        
         for idx, image_url in enumerate(df['image_url']):
             db_image = fetch_image_from_url(image_url)
             if db_image is None:
                 continue
 
-            try:
-                db_inputs = processor(images=db_image, return_tensors="pt")
-                db_image_features = model.get_image_features(**db_inputs)
+            
+            db_inputs = processor(images=db_image, return_tensors="pt")
+            db_inputs = {k: v.to(device) for k, v in db_inputs.items()}  
+            db_image_features = model.get_image_features(**db_inputs).to(device)  
 
-                similarity = torch.nn.functional.cosine_similarity(image_features, db_image_features).item()
-                if similarity > best_score:
-                    best_score = similarity
-                    best_match_idx = idx
-            except UnidentifiedImageError:
-                continue
+            
+            similarity = torch.nn.functional.cosine_similarity(image_features, db_image_features).item()
+
+            
+            if similarity > best_score:
+                best_score = similarity
+                best_match_idx = idx
 
+        
         if best_match_idx is None:
             return None, "Error: No valid image match found in the database.", None
 
+       
         best_match = df.iloc[best_match_idx]
         return process_best_match(best_match, language)