Update app.py

app.py

@@ -32,7 +32,7 @@ def get_colour(image_urls, category):
     for i in range(len(colourLabels)):
         colourLabels[i] = colourLabels[i] + " clothing: " + category
 
-    responses = pipe(image_urls, candidate_labels=colourLabels, device=device)
+    responses = pipe(image_urls, candidate_labels=colourLabels)
     # Get the most common colour
     mainColour = responses[0][0]['label'].split(" clothing:")[0]
 
@@ -46,7 +46,7 @@ def get_colour(image_urls, category):
         labels[i] = labels[i] + " clothing: " + category
 
     # Run pipeline in one go
-    responses = pipe(image_urls, candidate_labels=labels, device=device)
+    responses = pipe(image_urls, candidate_labels=labels)
     subColour = responses[0][0]['label'].split(" clothing:")[0]
 
     return subColour, mainColour, responses[0][0]['score']