Update app.py

app.py

@@ -72,7 +72,7 @@ if  num in [str(x) for x in list(range(10))]:
 
     with st.expander("parameter"):
         
-        beams = st.slider("num_beams",min_value=1,max_value=10,step=1,value=1,key=2)
+        beams = st.slider("num_beams",min_value=1,max_value=10,step=1,value=5,key=2)
         tokens = st.slider("min_new_tokens",min_value=0,max_value=50,step=1,value=20,key=3)
         top_p = st.slider("top_p",min_value=0.50,max_value=1.00,value=0.90,step=0.01)
         top_k = st.slider("top_k",min_value=5,max_value=50,value=30,step=1)
@@ -127,12 +127,9 @@ if  num in [str(x) for x in list(range(10))]:
                             )
             # output_list = []
             st.write(input)
-            for x in range(beams):
+            for x in range(5):
                 output_text = tokenizer.decode(output_ids[x], skip_special_tokens=True,
                                                     clean_up_tokenization_spaces=False)
                 st.write(output_text)
 
     
-        # temperature = st.slider("temperature",min_value=0.0,max_value=1.0,step=0.01,value=0.3,key=1)
-        # num_beams = st.slider("num_beams",min_value=1,max_value=5,step=1,value=1,key=2)
-        # min_new_tokens = st.slider("min_new_tokens",min_value=0,max_value=100,value=30,key=3)