added weights

pages/task3.py

@@ -29,12 +29,11 @@ button_style = """
 
 DEVICE = 'cpu' 
 
-# Загрузка пользовательской модели и токенизатора (замените на свои пути и модель)
-# model_path = "sberbank-ai/rugpt3small_based_on_gpt2"
-# tokenizer_path = "sberbank-ai/rugpt3small_based_on_gpt2"
 
-# model = GPT2LMHeadModel.from_pretrained(model_path).to(DEVICE)
-# tokenizer = GPT2Tokenizer.from_pretrained(tokenizer_path)
+tokenizer_path = "sberbank-ai/rugpt3small_based_on_gpt2"
+
+model = torch.load('srcs/gpt_weights.pth').to(DEVICE)
+tokenizer = GPT2Tokenizer.from_pretrained(tokenizer_path)
 
 st.markdown("""
     <style>