Update gpt.py

gpt.py

@@ -4,7 +4,7 @@ import torch
 import textwrap
 
 
-st.markdown('## Генерация текста GPT-моделью по пользовательскому prompt')
+st.markdown('## Генерация текста GPT-моделью')
 
 tokenizer = GPT2Tokenizer.from_pretrained('sberbank-ai/rugpt3small_based_on_gpt2')
 model = GPT2LMHeadModel.from_pretrained(
@@ -16,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([10, 5, 10])
+col1, col2, col3 = st.columns([4, 3, 10])
 
 with col1:
 
@@ -31,7 +31,7 @@ with col2:
 
 with col3:
 
-    prompt = st.text_input('Введите текст prompt:')
+    prompt = st.text_input('Введите текст:')
 
     if st.button('Сгенерировать текст'):