MARI-posa commited on
Commit
1fcea40
1 Parent(s): e310d7f

Update stri.py

Browse files
Files changed (1) hide show
  1. stri.py +1 -1
stri.py CHANGED
@@ -13,7 +13,7 @@ tokenizer = AutoTokenizer.from_pretrained(model_name)
13
  model = AutoModel.from_pretrained(model_name, output_hidden_states=True)
14
 
15
  # Загрузка датасета и аннотаций к книгам
16
- books = pd.read_csv('book_train.csv')
17
  books.dropna(inplace=True)
18
 
19
  books = books.reset_index(drop=True)
 
13
  model = AutoModel.from_pretrained(model_name, output_hidden_states=True)
14
 
15
  # Загрузка датасета и аннотаций к книгам
16
+ books = pd.read_csv('book_6000.csv')
17
  books.dropna(inplace=True)
18
 
19
  books = books.reset_index(drop=True)