Update general_bias_measurement.py

general_bias_measurement.py

@@ -21,13 +21,23 @@ BLIP_model = BlipForConditionalGeneration.from_pretrained("Salesforce/blip-image
 CLIP_model = CLIPModel.from_pretrained("openai/clip-vit-large-patch14").to(device)
 CLIP_processor = CLIPProcessor.from_pretrained("openai/clip-vit-large-patch14")
 
+#irrelevantWords = ['a', 'an', 'with', 'the', 'and', 'for', 'on', 'their', 'this', 'that', 'under', 'it', 'at', 'out',
+#                   'in', 'inside', 'outside', 'of', 'many', 'one', 'two', 'three', 'four', 'five', '-', 'with',
+#                   'six', 'seven', 'eight', 'none', 'ten', 'at', 'is', 'up', 'are', 'by', 'as', 'ts', 'there',
+#                   'like', 'bad', 'good', 'who', 'through', 'else', 'over', 'off', 'on', 'next',
+#                   'to', 'into', 'themselves', 'front', 'down', 'some', 'his', 'her', 'its', 'onto', 'eaten',
+#                   'each', 'other', 'most', 'let', 'around', 'them', 'while', 'another', 'from', 'above', "'",
+#                    '-', 'about', 'what', '', ' ', 'A', 'looks', 'has', 'background', 'behind' ]
+
+# Adding colors to limit
 irrelevantWords = ['a', 'an', 'with', 'the', 'and', 'for', 'on', 'their', 'this', 'that', 'under', 'it', 'at', 'out',
                    'in', 'inside', 'outside', 'of', 'many', 'one', 'two', 'three', 'four', 'five', '-', 'with',
                    'six', 'seven', 'eight', 'none', 'ten', 'at', 'is', 'up', 'are', 'by', 'as', 'ts', 'there',
                    'like', 'bad', 'good', 'who', 'through', 'else', 'over', 'off', 'on', 'next',
                    'to', 'into', 'themselves', 'front', 'down', 'some', 'his', 'her', 'its', 'onto', 'eaten',
                    'each', 'other', 'most', 'let', 'around', 'them', 'while', 'another', 'from', 'above', "'",
-                    '-', 'about', 'what', '', ' ', 'A', 'looks', 'has', 'background', 'behind' ]
+                   '-', 'about', 'what', '', ' ', 'A', 'looks', 'has', 'background', 'behind',
+                   'red', 'blue', 'green', 'brown', 'yellow', 'purple', 'orange', 'white', 'black' ]
 
 # Variables for the LLM
 maxLength = 10