Update app.py

app.py

@@ -171,12 +171,12 @@ def get_predicted_attributes(image_urls, category):
 
         # If attribute is details, then remove the obtained label from the values, and get the next most common
         if attribute == "details":
-            print(values_formatted, result[0].strip() + f", clothing: {category}")
-            values_formatted.remove(result[0].strip() + f", clothing: {category}")
+            print(values_formatted, result[0] + f", clothing: {category}")
+            values_formatted.remove(result[0] + f", clothing: {category}")
             responses = pipe(image_urls, candidate_labels=values_formatted)
             i = 0
             print("Responses: ", responses)
-            while len(responses) > 0 and responses[0][0]['score'] > 0.8 and i < 2:
+            while len(responses) > 0 and responses[0][0]['score'] > 0.6 and i < 2:
                 result = [response[0]['label'].split(", clothing:")[0] for response in responses]
                 common_result.append(Counter(result).most_common(1))