Update app.py

app.py

@@ -9,19 +9,20 @@ def loadModels():
   _tokenizer = PreTrainedTokenizerFast.from_pretrained(repository)
   
   print("Loaded :)")
-  
   return _model, _tokenizer
-
+  
 model, tokenizer = loadModels()
 
 lit.title("성경말투 생성기")
 lit.caption("적당한 길이의 한 문장을 넣었을 때 가장 좋은 결과가 나옵니다.")
 lit.caption("https://github.com/rycont/kobart-biblify")
 
-text_input = lit.text_area("문장 입력")
-
 MAX_LENGTH = 128
 
+with lit.form("GEN"):
+  text_input = lit.text_area("문장 입력")
+  submitted = lit.form_submit_button("생성")
+
 def biblifyWithBeams(beam, tokens, attention_mask):
   generated = model.generate(
     input_ids = torch.Tensor([ tokens ]).to(torch.int64),
@@ -30,8 +31,8 @@ def biblifyWithBeams(beam, tokens, attention_mask):
     max_length = MAX_LENGTH,
     eos_token_id=tokenizer.eos_token_id,
     bad_words_ids=[[tokenizer.unk_token_id]]
-   )[0]
-   
+  )[0]
+  
   return tokenizer.decode(
     generated,
   ).replace('<s>', '').replace('</s>', '')
@@ -40,9 +41,10 @@ if len(text_input.strip()) > 0:
   print(text_input)
   
   text_input = "<s>" + text_input + "</s>"
-  tokens = tokenizer.encode(text_input)
   
+  tokens = tokenizer.encode(text_input)
   tokenLength = len(tokens)
+  
   attentionMasks = [ 1 ] * tokenLength + [ 0 ] * (MAX_LENGTH - tokenLength)
   tokens = tokens + [ tokenizer.pad_token_id ] * (MAX_LENGTH - tokenLength)
   
@@ -53,17 +55,16 @@ if len(text_input.strip()) > 0:
       i + 1,
       tokens,
       attentionMasks
-    )
-    
-    if generated in results:
-      print("중복됨")
-      continue
-      
-    results.append(generated)
-    with lit.expander(str(len(results)) + "번째 결과 (" + str(i +1) + ")", True):
-      lit.write(generated)
-      print(generated)
-      lit.caption(
-        "및 " + str(5 - len(results)) + " 개의 중복된 결과")
-
-
+     )
+     
+     if generated in results:
+       print("중복됨")
+       continue
+       
+     results.append(generated)
+     
+     with lit.expander(str(len(results)) + "번째 결과 (" + str(i +1) + ")", True):
+       lit.write(generated)
+       print(generated)
+     
+     lit.caption("및 " + str(5 - len(results)) + " 개의 중복된 결과")