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(image_urls, candidate_labels=values, device=device)
+    responses = pipe(image_urls, candidate_labels=values)
     most_common, score = get_most_common_label(responses)
     common_result.append(most_common)
 
@@ -179,7 +179,7 @@ def get_predicted_attributes(image_urls, category):
         # Process additional details labels if the score is higher than 0.8
         for _ in range(2):
             values = [value for value in values if value != f"{most_common}, clothing: {category}"]
-            responses = pipe(product['Images'].values[0], candidate_labels=values, device=device)
+            responses = pipe(image_urls, candidate_labels=values)
             most_common, score = get_most_common_label(responses)
             if score > 0.8:
                 common_result.append(most_common)