Update app.py

app.py

@@ -55,7 +55,7 @@ def generate_text(model_GPT, tokenizer_GPT, prompt):
 
 # Streamlit приложение
 def main():
-    model_GPT, tokenizer_GPT = load_gpt()
+    model_GPT, tokenizer_GPT = load_gpt_base()
     st.write("""
     # Fine-tuned GPT-2 for New Language with Custom Tokenizer
     """)
@@ -84,8 +84,8 @@ def main():
     # Обработка события нажатия кнопки
     if generate_button1:
     # Вывод сгенерированного текста
-        #generated_text = generate_text(model_GPT, tokenizer_GPT, prompt)
-        generated_text = 'test'
+        generated_text = generate_text(model_GPT, tokenizer_GPT, prompt)
+        #generated_text = 'test'
         st.subheader("Completed prompt:")
         st.write(generated_text)