Update gpt.py

gpt.py

@@ -3,6 +3,9 @@ import streamlit as st
 import torch
 import textwrap
 
+
+st.markdown('## Генерация текста GPT-моделью по пользовательскому prompt')
+
 tokenizer = GPT2Tokenizer.from_pretrained('sberbank-ai/rugpt3small_based_on_gpt2')
 model = GPT2LMHeadModel.from_pretrained(
     'sberbank-ai/rugpt3small_based_on_gpt2',
@@ -13,7 +16,7 @@ model = GPT2LMHeadModel.from_pretrained(
 model.load_state_dict(torch.load('modelgpt.pt', map_location=torch.device('cpu')))
 
 
-col1, col2, col3 = st.columns([9, 1, 9])
+col1, col2, col3 = st.columns([10, 5, 10])
 
 with col1: