Update app.py

app.py

@@ -157,11 +157,13 @@ def get_most_common_label(responses):
 def get_predicted_attributes(image_urls, category):
     # Assuming ATTRIBUTES_DICT and pipe are defined outside this function
     attributes = list(ATTRIBUTES_DICT.get(category, {}).keys())
-
+    print(category, attributes)
+    
     # Mapping of possible values per attribute
     common_result = []
     for attribute in attributes:
         values = ATTRIBUTES_DICT.get(category, {}).get(attribute, [])
+        print(attribute, values)
 
         if len(values) == 0:
             continue
@@ -174,6 +176,7 @@ def get_predicted_attributes(image_urls, category):
     responses = pipe(image_urls, candidate_labels=values)
     most_common, score = get_most_common_label(responses)
     common_result.append(most_common)
+    print(common_result)
 
     if attribute == "details":
         # Process additional details labels if the score is higher than 0.8