Update app.py

app.py

@@ -68,7 +68,7 @@ def Get_Densefeature(image, candidate_labels):
 
     with torch.no_grad():
         dense_image_feature = model.get_image_dense_features(image_input)
-        captions = candidate_labels
+        captions = [candidate_labels[0]]
         caption_input = torch.tensor(tokenizer(captions, max_length=77, padding="max_length", truncation=True).input_ids, dtype=torch.long, device=device)
         text_feature = model.get_text_features(caption_input,walk_short_pos=True)
         text_feature = text_feature / text_feature.norm(p=2, dim=-1, keepdim=True)