Update app_pages/page3_gpt_model.py

app_pages/page3_gpt_model.py

@@ -6,9 +6,10 @@ def run():
     st.title('GPT Text Generation')
     prompt_text = st.text_area("Input Text", "Type here...")
     length = st.slider("Length of Generated Text", min_value=50, max_value=500, value=200)
-    temperature = st.slider("Temperature", min_value=0.1, max_value=1.0, value=0.7, step=0.1)
+    temperature = st.slider("Temperature", min_value=0.1, max_value=2.0, value=0.7, step=0.1)
+    beams = st.slider("Number of Generations", min_value=2, max_value=10, value=4, step=1)
 
     if st.button('Generate Text'):
         with st.spinner('Generating...'):
-            generated_text = generate_text(prompt_text, length, temperature)
+            generated_text = generate_text(prompt_text, length, temperature, beams)
             st.text_area("Generated Text", generated_text, height=250)