MARI-posa commited on
Commit
3ce4e56
1 Parent(s): 1b0b406

Update stri.py

Browse files
Files changed (1) hide show
  1. stri.py +1 -1
stri.py CHANGED
@@ -17,7 +17,7 @@ tokenizer = AutoTokenizer.from_pretrained(model_name)
17
  model = AutoModel.from_pretrained(model_name, output_hidden_states=True)
18
 
19
  # Загрузка датасета и аннотаций к книгам
20
- books = pd.read_csv('all+.csv')
21
  books.dropna(inplace=True)
22
 
23
  books = books[books['annotation'].apply(lambda x: len(x.split()) >= 40)]
 
17
  model = AutoModel.from_pretrained(model_name, output_hidden_states=True)
18
 
19
  # Загрузка датасета и аннотаций к книгам
20
+ books = pd.read_csv('all+++.csv')
21
  books.dropna(inplace=True)
22
 
23
  books = books[books['annotation'].apply(lambda x: len(x.split()) >= 40)]