Update app.py

app.py

@@ -171,7 +171,7 @@ def get_predicted_attributes(image_urls, category):
     values = [f"{attribute}: {value.strip()}, clothing: {category}" for value in values]
 
     # Get the predicted values for the attribute
-    responses = pipe(product['Images'].values[0], candidate_labels=values, device=device)
+    responses = pipe(image_urls, candidate_labels=values, device=device)
     most_common, score = get_most_common_label(responses)
     common_result.append(most_common)