Update app.py

app.py

@@ -69,6 +69,12 @@ if  num in [str(x) for x in list(range(10))]:
 
     #盤面表示
     s = st.selectbox(label="手数を選択",options=te)
+
+    with st.expander("parameter"):
+        temp = st.slider("temperature",min_value=0.0,max_value=1.0,step=0.01,value=0.3,key=1)
+        beams = st.slider("num_beams",min_value=1,max_value=5,step=1,value=1,key=2)
+        tokens = st.slider("min_new_tokens",min_value=0,max_value=50,value=20,key=3
+    
     reload = st.button('盤面生成',key=0)
     if s in te and reload == True:
         reload = False
@@ -105,6 +111,9 @@ if  num in [str(x) for x in list(range(10))]:
             output_ids = model.generate(input_ids=tokenized_inputs,
                             max_length=512,
                             repetition_penalty=10.0, # 同じ文の繰り返しへのペナルティ
+                            temperature = temp,
+                            num_beams = beams,
+                            min_new_tokens = tokens,
                             )
         
             output_text = tokenizer.decode(output_ids[0], skip_special_tokens=True,