Update app.py

app.py

@@ -69,7 +69,7 @@ if  num in [str(x) for x in list(range(10))]:
 
     #盤面表示
     s = st.selectbox(label="手数を選択",options=te)
-    reload = st.button('生成',key=0)
+    reload = st.button('盤面生成',key=0)
     if s in te and reload == True:
         reload = False
         idx = te.index(s)
@@ -99,24 +99,25 @@ if  num in [str(x) for x in list(range(10))]:
         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)
         #st.write(idx,"入力",input)
+        reload2 = st.button('生成',key=4)
+        if reload2:
+            with st.spinner("推論中です..."):
+                tokenized_inputs = tokenizer.encode(
+                                    input, max_length= 512, truncation=True, 
+                                    padding="max_length", return_tensors="pt"
+                                )
         
-        with st.spinner("推論中です..."):
-            tokenized_inputs = tokenizer.encode(
-                                input, max_length= 512, truncation=True, 
-                                padding="max_length", return_tensors="pt"
+                output_ids = model.generate(input_ids=tokenized_inputs,
+                            max_length=512,
+                            repetition_penalty=10.0, # 同じ文の繰り返しへのペナルティ
+                            temperature = temperature,
+                            num_beams = num_beams,
+                            min_new_tokens = min_new_tokens,
                             )
-    
-            output_ids = model.generate(input_ids=tokenized_inputs,
-                        max_length=512,
-                        repetition_penalty=10.0, # 同じ文の繰り返しへのペナルティ
-                        temperature = temperature,
-                        num_beams = num_beams,
-                        min_new_tokens = min_new_tokens,
-                        )
-    
-            output_text = tokenizer.decode(output_ids[0], skip_special_tokens=True,
-                                            clean_up_tokenization_spaces=False)
-            st.write(output_text)
+        
+                output_text = tokenizer.decode(output_ids[0], 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)