Update app.py

app.py

@@ -52,7 +52,7 @@ def process(caption, category, asr, ocr):
         tags_dis = np.load('./tag_data/tags_dis.npy')
 
         candidate_tags = [_[0] for _ in tag_count]
-        encoder = SentenceTransformer("hfl/chinese-roberta-wwm-ext-large")
+        encoder = SentenceTransformer("hfl/chinese-roberta-wwm-ext-large", device='cuda')
         candidate_tags_embed = encoder.encode(candidate_tags)
         candidate_tags_dis = [np.sqrt(np.dot(_, _.T)) for _ in candidate_tags_embed]