Update app.py

app.py

@@ -165,22 +165,22 @@ if input3==3:
 #print(cond)
 
 
-tokenizer = text_tokenizer
-title_token_ids=tokenizer.encode(st.session_state.title, add_special_tokens=False)
-condition_token_ids=cond_tokenizer.EncodeAsIds(st.session_state.aspect+" "+st.session_state.cat)
-fact_token_ids=tokenizer.encode(st.session_state.fact, add_special_tokens=False)
-
-e = Example(title_token_ids, condition_token_ids, fact_token_ids)
 
+if st.button('result'):
 
-dm = KobeDataModule(
+    tokenizer = text_tokenizer
+    title_token_ids=tokenizer.encode(st.session_state.title, add_special_tokens=False)
+    condition_token_ids=cond_tokenizer.EncodeAsIds(st.session_state.aspect+" "+st.session_state.cat)
+    fact_token_ids=tokenizer.encode(st.session_state.fact, add_special_tokens=False)
+    e = Example(title_token_ids, condition_token_ids, fact_token_ids)
+    
+    dm = KobeDataModule(
         [e],
         args.text_vocab_path,
         args.max_seq_len,
         1,
         1,
     )
-if st.button('result'):
     
     for d in dm.test_dataloader():
         st.write(st.session_state.title)