Replace tags

app.py

@@ -165,7 +165,7 @@ def get_predicted_attributes(image_urls, category):
 
         # Get the predicted values for the attribute
         responses = pipe(image_urls, candidate_labels=values_formatted)
-        result = [response[0]['label'].split(", clothing:")[0] for response in responses]
+        result = [response[0]['label'].split(", clothing:")[0].replace("collar", "colartype").replace("sleeve length", "sleevelength").replace("fabric", "fabricstyle") for response in responses]
 
         # If attribute is details, then get the top 2 most common labels
         if attribute_formatted == "details":