Update pages/gpt.py

pages/gpt.py

@@ -1,15 +1,13 @@
 from transformers import GPT2LMHeadModel, GPT2Tokenizer
 
-tokenizer = GPT2Tokenizer.from_pretrained('gpt2')
-model = GPT2LMHeadModel.from_pretrained('model.ph')
 
-
-import streamlit as st
-
-prompt = st.text_input('Введите текст prompt:')
-length = st.slider('Длина генерируемой последовательности:', 10, 1000, 50)
-num_samples = st.slider('Число генераций:', 1, 10, 1)
-temperature = st.slider('Температура:', 0.1, 1.0, 0.5)
+model = GPT2LMHeadModel.from_pretrained(
+    'sberbank-ai/rugpt3small_based_on_gpt2',
+    output_attentions = False,
+    output_hidden_states = False,
+)
+# Вешаем сохраненные веса на нашу модель
+model.load_state_dict(torch.load('model.pt'))
 
 
 import torch