Update pages/GPT.py

pages/GPT.py

@@ -1,4 +1,17 @@
 import streamlit as st
+from models.generate_text import generate_text
 
-x = st.slider('Select a value')
-st.write(x, 'squared is', x * x)
+st.title("Генерация текста GPT-моделью")
+
+prompt = st.text_area("Введите Ваш текст:")
+length = st.slider("Длинна текста", min_value=50, max_value=1000, value=200)
+num_generations = st.number_input("Число генераций", min_value=1, max_value=10, value=1)
+temperature = st.slider("Температура", min_value=0.1, max_value=1.0, value=0.7)
+top_k = st.number_input("Top-k", min_value=0, max_value=50, value=0)
+top_p = st.slider("Top-p", min_value=0.0, max_value=1.0, value=0.9)
+
+if st.button("Генерация текста"):
+    outputs = generate_text(prompt, length, num_generations, temperature, top_k, top_p)
+    
+    for i, output in enumerate(outputs):        
+        st.write(output)