Update app.py

app.py

@@ -94,16 +94,16 @@ class Pix2PixDataset(torch.utils.data.Dataset):
         original = original_img.convert('RGB')
         target = target_img.convert('RGB')
         
-        # Extract the filename from the original image's path (assuming it has a 'filename' field or path)
-        original_img_path = self.originals[idx]['image'].filename  # Assuming it has this attribute
+        # Extract the filename from the original image's path
+        original_img_path = original_img.filename  # Ensure this is the correct way to get the path
         original_img_filename = os.path.basename(original_img_path)
-        
+    
         # Match the image filename with the `image_path` column in the CSV
         matched_row = self.data[self.data['image_path'].str.contains(original_img_filename)]
         
         if matched_row.empty:
             raise ValueError(f"No matching entry found in the CSV for image {original_img_filename}")
-
+    
         # Get the prompts from the matched row
         original_prompt = matched_row['original_prompt'].values[0]
         enhanced_prompt = matched_row['enhanced_prompt'].values[0]