Update app.py

app.py

@@ -8,7 +8,7 @@ from torch import tensor
 
 import joblib
 from dataclasses import dataclass
-from transformers import AutoTokenizer, AutoModel, AutoModelForSequenceClassification, GPT2LMHeadModel, GPT2Tokenizer
+from transformers import AutoTokenizer, AutoModel, AutoModelForSequenceClassification, GPT2LMHeadModel, GPT2Tokenizer, GPT2Config
 import json
 
 from preprocessing import predict_review, data_preprocessing_hard
@@ -191,6 +191,7 @@ elif selected_model == "Генерация текста GPT-моделью по
     model = GPT2LMHeadModel.from_pretrained(model_name_or_path).to(DEVICE)
     model = GPT2LMHeadModel(config)
     model.load_state_dict(torch.load('model_dict.pt', map_location=device))
+    model.eval()
     
     if st.button('Сделать гороскоп'):
         start_time = time.time()