Update app.py

app.py

@@ -42,7 +42,7 @@ with open(attributes_file_path, 'r') as file:
 COLOURS_DICT = color_data['color_mapping']
 ATTRIBUTES_DICT = attributes_data['attribute_mapping']
 
-DETAILS_THRESHOLD = 0.8 # This is how high the total score of an additional detail attribute should be for it to be included
+DETAILS_THRESHOLD = 0.4 # This is how high the total score of an additional detail attribute should be for it to be included (number scales up linearly with more images)
 
 
 def shot(input, category, level):
@@ -183,7 +183,7 @@ def get_predicted_attributes(image_urls, category):
                 values = [value for value in values if value != f"{most_common}, clothing: {category}"]
                 responses = pipe(image_urls, candidate_labels=values)
                 most_common, score = get_most_common_label(responses)
-                if score > DETAILS_THRESHOLD:
+                if score > DETAILS_THRESHOLD * len(image_urls):
                     common_result.append(most_common)
 
     # Convert common_result into a dictionary