Spaces:

mechtnet
/

music_text

Sleeping

mechtnet commited on Dec 21, 2024

Commit

43ee97d

verified ·

1 Parent(s): 9670299

Update text_analysis.py

Files changed (1) hide show

text_analysis.py CHANGED Viewed

@@ -1,8 +1,8 @@
 import torch
 from transformers import LongformerModel, LongformerTokenizerFast
-model = LongformerModel.from_pretrained('kazzand/ru-longformer-large-4096')
-tokenizer = LongformerTokenizerFast.from_pretrained('kazzand/ru-longformer-large-4096')
 def get_cls_embedding(text, model, tokenizer, device='cuda'):
     model.to(device)

 import torch
 from transformers import LongformerModel, LongformerTokenizerFast
+model = LongformerModel.from_pretrained('kazzand/ru-longformer-base-4096')
+tokenizer = LongformerTokenizerFast.from_pretrained('kazzand/ru-longformer-base-4096')
 def get_cls_embedding(text, model, tokenizer, device='cuda'):
     model.to(device)