Update app.py

app.py

@@ -186,10 +186,8 @@ elif selected_model == "Генерация текста GPT-моделью по
     user_text_input = st.text_area('Введите информацию о себе для формиорования гороскопа:')
 
     # GPT2
-    model_name_or_path = "sberbank-ai/rugpt3small_based_on_gpt2"
-    tokenizer = GPT2Tokenizer.from_pretrained(model_name_or_path)
-    model = GPT2LMHeadModel.from_pretrained(model_name_or_path).to(device)
-    model.load_state_dict(torch.load('model_dict.pt', map_location=device))
+    model = GPT2LMHeadModel.from_pretrained("model_pretra.zip").to(device)
+    tokenizer = GPT2Tokenizer.from_pretrained("tokenizer.pt.zip")
     
     if st.button('Сделать гороскоп'):
         start_time = time.time()