Update app.py

app.py

@@ -24,7 +24,7 @@ model = GPT2LMHeadModel.from_pretrained(
  )
 
 # # Вешаем сохраненные веса на нашу модель
- model.load_state_dict(torch.load('model_history.pt'))
+model.load_state_dict(torch.load('model_history.pt'))
 # Функция для генерации текста
 def generate_text(prompt):
     # Преобразование входной строки в токены
@@ -39,7 +39,8 @@ 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("""
@@ -62,5 +63,4 @@ def main():
 
 if __name__ == "__main__":
     main()
-   # Создание кнопки "Сгенерировать"
-    generate_button = st.button("За работу!")
+