Update app.py

app.py

@@ -52,15 +52,15 @@ def main():
 
     st.write("Fine-tuned GPT-2 for New Language with Custom Tokenizer")
 
-    # Блок для генерации текста с GPT-2
-    st.subheader("GPT-2 Text Generation")
-    prompt_gpt2 = st.text_area("Введите фразу для GPT-2 генерации:", value="В средние века")
-    generate_button_gpt2 = st.button("Сгенерировать текст с GPT-2")
+    # # Блок для генерации текста с GPT-2
+    # st.subheader("GPT-2 Text Generation")
+    # prompt_gpt2 = st.text_area("Введите фразу для GPT-2 генерации:", value="В средние века")
+    # generate_button_gpt2 = st.button("Сгенерировать текст с GPT-2")
 
-    if generate_button_gpt2:
-        generated_text_gpt2 = generate_text(model_gpt2, tokenizer_gpt2, prompt_gpt2, max_len, temperature)
-        st.subheader("Результат генерации GPT-2:")
-        st.write(generated_text_gpt2)
+    # if generate_button_gpt2:
+    #     generated_text_gpt2 = generate_text(model_gpt2, tokenizer_gpt2, prompt_gpt2, max_len, temperature)
+    #     st.subheader("Результат генерации GPT-2:")
+    #     st.write(generated_text_gpt2)
 
     # Блок для генерации текста с кастомной моделью
     st.subheader("Custom Model Text Generation")