Bugfix length of text prompts

app.py

@@ -78,7 +78,7 @@ with torch.no_grad():
     hair_text_features = hair_text_features.cpu()
 
 
-def get_cosine_similarities(image_features, text_features):
+def get_cosine_similarities(image_features, text_features, text_prompts):
     cosine_simlarities = softmax(
         (text_features @ image_features.cpu().T)
         .squeeze()
@@ -96,10 +96,10 @@ def predict_fn(input_img):
     with torch.no_grad():
         image_features = clip_model.encode_image(image)
         base_body_cosine_simlarities = get_cosine_similarities(
-            image_features, all_text_features
+            image_features, all_text_features, text_prompts
         )
         hair_cosine_simlarities = get_cosine_similarities(
-            image_features, hair_text_features
+            image_features, hair_text_features, hair_text_prompts
         )
         # logger.info(f"cosine_simlarities shape: {cosine_simlarities.shape}")
         logger.info(f"cosine_simlarities: {base_body_cosine_simlarities}")