lllchenlll commited on
Commit
45fab91
1 Parent(s): 3eb61be

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -41,7 +41,7 @@ def process(api, caption, category, asr, ocr):
41
  res += ans.split('、')
42
 
43
  tag_count = Counter(res)
44
- tag_count = sorted(tag_count.items(), key=lambda x: x[1], reverse=True)[:10]
45
 
46
  tags_embed = np.load('./tag_data/tags_embed.npy')
47
  tags_dis = np.load('./tag_data/tags_dis.npy')
 
41
  res += ans.split('、')
42
 
43
  tag_count = Counter(res)
44
+ tag_count = sorted(tag_count.items(), key=lambda x: x[1], reverse=True)
45
 
46
  tags_embed = np.load('./tag_data/tags_embed.npy')
47
  tags_dis = np.load('./tag_data/tags_dis.npy')