Update app.py

app.py

@@ -75,7 +75,7 @@ def get_predicted_attributes(image_urls, category):
     common_result = []
     for attribute in attributes:
         # values = get_attribute_values(attribute, category)
-        values = list(ATTRIBUTES_DICT.get(category,{}).get(attribute,{}).keys())
+        values = ATTRIBUTES_DICT.get(category,{}).get(attribute,[])
 
         if len(values) == 0:
             continue