Update pages/gpt_v1.py

pages/gpt_v1.py

@@ -2,6 +2,8 @@ from transformers import GPT2LMHeadModel, GPT2Tokenizer
 import torch
 import streamlit as st
 
+st.title('Генерация текста GPT-моделью по пользовательскому prompt')
+
 model = GPT2LMHeadModel.from_pretrained(
     'sberbank-ai/rugpt3small_based_on_gpt2',
     output_attentions = False,
@@ -12,14 +14,6 @@ tokenizer = GPT2Tokenizer.from_pretrained('sberbank-ai/rugpt3small_based_on_gpt2
 # Вешаем сохраненные веса на нашу модель
 model.load_state_dict(torch.load('model.pt', map_location=torch.device('cpu')))
 
-col1, col2, col3 = st.columns([1,8,1])
-#col1, col2 = st.columns(2)
-
-### Гистограмма total_bill
-with col2:
-# Веб-приложение с использованием Streamlit
-    st.title('Генерация текста GPT-моделью по пользовательскому prompt')
-
 prompt = st.text_input('Введите текст prompt:')
 length = st.slider('Длина генерируемой последовательности:', 1, 256, 16)
 num_samples = st.slider('Число генераций:', 1, 6, 1)