Update app.py

app.py

@@ -53,41 +53,52 @@ def generate_text(model_GPT, tokenizer_GPT, prompt):
 
     return generated_text
 
+# Streamlit приложение
+import streamlit as st
+
+# Пример функции для загрузки модели GPT-2 и токенизатора
+def load_gpt_base():
+    model = None  # Здесь будет ваша модель
+    tokenizer = None  # Здесь будет ваш токенизатор
+    return model, tokenizer
+
+# Пример функции генерации текста
+def generate_text(model, tokenizer, prompt):
+    return f"Generated text based on: {prompt}"  # Пример генерации текста
+
 # Streamlit приложение
 def main():
     model_GPT, tokenizer_GPT = load_gpt_base()
+
     st.write("""
     # Fine-tuned GPT-2 for New Language with Custom Tokenizer
     """)
 
-    # Ввод строки пользователем
-    prompt = st.text_area("Какую фразу нужно продолжить:", value="В средние века")
+    # Ввод строки пользователем для первой задачи
+    prompt = st.text_area("Введите фразу для первой генерации:", value="В средние века")
+
+    # Создание кнопки для первой генерации
+    generate_button = st.button("Сгенерировать текст 1")
 
-    # # Генерация текста по введенной строке
-    # generated_text = generate_text(prompt)
-    # Создание кнопки "Сгенерировать"
-    generate_button = st.button("Complete!")
-    # Обработка события нажатия кнопки
+    # Обработка события нажатия первой кнопки
     if generate_button:
-    # Вывод сгенерированного текста
-        #generated_text = generate_text(model_GPT, tokenizer_GPT, prompt)
-        generated_text = 'test'
-        st.subheader("Completed prompt:")
+        generated_text = generate_text(model_GPT, tokenizer_GPT, prompt)
+        st.subheader("Результат первой генерации:")
         st.write(generated_text)
 
-        # Ввод строки пользователем
-    prompt1 = st.text_area("Какую фразу нужно продолжить:", value="В средние века")
-        # # Генерация текста по введенной строке
-    # generated_text = generate_text(prompt)
-    # Создание кнопки "Сгенерировать"
-    generate_button1 = st.button("Complete!")
-    # Обработка события нажатия кнопки
+    # Ввод строки пользователем для второй задачи
+    prompt1 = st.text_area("Введите фразу для второй генерации:", value="Когда-то давно")
+
+    # Создание кнопки для второй генерации
+    generate_button1 = st.button("Сгенерировать текст 2")
+
+    # Обработка события нажатия второй кнопки
     if generate_button1:
-    # Вывод сгенерированного текста
-        generated_text = generate_text(model_GPT, tokenizer_GPT, prompt)
-        #generated_text = 'test'
-        st.subheader("Completed prompt:")
-        st.write(generated_text)
+        generated_text1 = generate_text(model_GPT, tokenizer_GPT, prompt1)
+        st.subheader("Результат второй генерации:")
+        st.write(generated_text1)
+
+
 
 if __name__ == "__main__":
     main()