Update app.py

app.py

@@ -44,8 +44,7 @@ def generate_text(prompt):
     generated_text = tokenizer.decode(output[0], skip_special_tokens=True)
 
     return generated_text
-   # Создание кнопки "Сгенерировать"
-generate_button = st.button("За работу!")
+
 # Streamlit приложение
 def main():
     st.write("""
@@ -55,12 +54,14 @@ def main():
     # Ввод строки пользователем
     prompt = st.text_area("Какую фразу нужно продолжить:", value="В средние века на руси")
 
-    # Генерация текста по введенной строке
-    generated_text = generate_text(prompt)
- 
+    # # Генерация текста по введенной строке
+    # generated_text = generate_text(prompt)
+    # Создание кнопки "Сгенерировать"
+    generate_button = st.button("За работу!")
     # Обработка события нажатия кнопки
     if generate_button:
     # Вывод сгенерированного текста
+        generated_text = generate_text(prompt)
         st.subheader("Продолжение:")
         st.write(generated_text)