Update app.py

app.py

@@ -77,31 +77,57 @@ def shot(input, category, level):
 
 
 
+# @spaces.GPU  
+# def get_colour(image_urls, category):
+#     colourLabels = list(COLOURS_DICT.keys())
+#     for i in range(len(colourLabels)):
+#         colourLabels[i] = colourLabels[i] + " clothing: " + category
+
+#     responses = pipe(image_urls, candidate_labels=colourLabels)
+#     # Get the most common colour
+#     mainColour = responses[0][0]['label'].split(" clothing:")[0]
+
+
+#     if mainColour not in COLOURS_DICT:
+#         return None, None, None
+
+#     # Add category to the end of each label
+#     labels = COLOURS_DICT[mainColour]
+#     for i in range(len(labels)):
+#         labels[i] = labels[i] + " clothing: " + category
+
+#     # Run pipeline in one go
+#     responses = pipe(image_urls, candidate_labels=labels)
+#     subColour = responses[0][0]['label'].split(" clothing:")[0]
+
+#     return subColour, mainColour, responses[0][0]['score']
 @spaces.GPU  
 def get_colour(image_urls, category):
     colourLabels = list(COLOURS_DICT.keys())
     for i in range(len(colourLabels)):
         colourLabels[i] = colourLabels[i] + " clothing: " + category
 
+    print("Colour Labels:", colourLabels)  # Debug: Print colour labels
+    print("Image URLs:", image_urls)       # Debug: Print image URLs
+
     responses = pipe(image_urls, candidate_labels=colourLabels)
-    # Get the most common colour
     mainColour = responses[0][0]['label'].split(" clothing:")[0]
 
-
     if mainColour not in COLOURS_DICT:
         return None, None, None
 
-    # Add category to the end of each label
     labels = COLOURS_DICT[mainColour]
     for i in range(len(labels)):
         labels[i] = labels[i] + " clothing: " + category
 
-    # Run pipeline in one go
+    print("Labels for pipe:", labels)  # Debug: Confirm labels are correct
     responses = pipe(image_urls, candidate_labels=labels)
     subColour = responses[0][0]['label'].split(" clothing:")[0]
 
     return subColour, mainColour, responses[0][0]['score']
 
+
+
 @spaces.GPU  
 def get_predicted_attributes(image_urls, category):
     # Assuming ATTRIBUTES_DICT and pipe are defined outside this function