Update app.py
Browse files
app.py
CHANGED
@@ -10,7 +10,6 @@ import joblib
|
|
10 |
from dataclasses import dataclass
|
11 |
from transformers import AutoTokenizer, AutoModel, AutoModelForSequenceClassification, GPT2LMHeadModel, GPT2Tokenizer
|
12 |
import json
|
13 |
-
import os
|
14 |
|
15 |
from preprocessing import predict_review, data_preprocessing_hard
|
16 |
from model_lstm import LSTMClassifier
|
@@ -188,9 +187,8 @@ elif selected_model == "Генерация текста GPT-моделью по
|
|
188 |
|
189 |
# GPT2
|
190 |
model_path = "model.safetensors"
|
191 |
-
|
192 |
-
|
193 |
-
tokenizer = GPT2Tokenizer.from_pretrained(model_path, token=huggingface_token)
|
194 |
|
195 |
if st.button('Сделать гороскоп'):
|
196 |
start_time = time.time()
|
|
|
10 |
from dataclasses import dataclass
|
11 |
from transformers import AutoTokenizer, AutoModel, AutoModelForSequenceClassification, GPT2LMHeadModel, GPT2Tokenizer
|
12 |
import json
|
|
|
13 |
|
14 |
from preprocessing import predict_review, data_preprocessing_hard
|
15 |
from model_lstm import LSTMClassifier
|
|
|
187 |
|
188 |
# GPT2
|
189 |
model_path = "model.safetensors"
|
190 |
+
model = GPT2LMHeadModel.from_pretrained(model_path)
|
191 |
+
tokenizer = GPT2Tokenizer.from_pretrained(model_path)
|
|
|
192 |
|
193 |
if st.button('Сделать гороскоп'):
|
194 |
start_time = time.time()
|