o i-se@sXddlZddlZddlmZmZddlmZddlZddl m Z ddddZ d d Z dS) N) AutoTokenizer AutoModel)LogisticRegression)timeuНейтральныйuПоложительныйuОтрицательный)rc st}td}td||dddd}td ifdd|D}Wdn1s4wY|jdddddf}tjj |}| }t}td}||}t|d}t} | |} d |d | d d S)Nzcointegrated/rubert-tiny2Tpt)padding truncationreturn_tensorscsi|] \}}||jqS)todevice).0kvmodelr g/Users/Anastasia/ds_bootcamp/Проекты ds-phase-2 /03_nlp_lstm_project_streamlit_app/model/bert.py sz#preprocess_bert..rzmodel/lr_weights.pklz***u-***, время предсказания: ***z.4fu сек***.r )rrfrom_pretrainedrtorchno_graditemslast_hidden_statenn functional normalizedetachcpunumpyrjoblibloadpredictdict) text start_time tokenizert model_output embeddingslrpredicted_labelpredicted_label_textend_timeinference_timer rrpreprocess_bert s"   "   r0) r streamlitst transformersrrsklearn.linear_modelrr!rr$r0r r r rs