Update app.py

app.py

@@ -85,7 +85,7 @@ def get_predicted_attributes(image_urls, category):
         values = [f"{attribute}: {value}, 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)
         result = [response[0]['label'].split(", clothing:")[0] for response in responses]
 
         # If attribute is details, then get the top 2 most common labels