Update app.py

app.py

@@ -186,10 +186,12 @@ elif selected_model == "Генерация текста GPT-моделью по
     user_text_input = st.text_area('Введите информацию о себе для формиорования гороскопа:')
 
     # GPT2
-    model_path = "my_model/"
-    model = GPT2LMHeadModel.from_pretrained(model_path)
-    tokenizer = GPT2Tokenizer.from_pretrained(model_path)
-
+    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 = GPT2LMHeadModel(config)
+    model.load_state_dict(torch.load('model_dict.pt', map_location=device))
+    
     if st.button('Сделать гороскоп'):
         start_time = time.time()
         input_ids = tokenizer.encode(user_text_input, return_tensors="pt").to(device)