snoop2head commited on
Commit
d19be82
·
1 Parent(s): ec0b50d

remove success messages

Browse files
Files changed (1) hide show
  1. app.py +1 -11
app.py CHANGED
@@ -25,15 +25,6 @@ src_text = st.text_area(
25
  )
26
  print(src_text)
27
 
28
- # num_beams = st.sidebar.slider(
29
- # "Num Beams", min_value=5, max_value=10, value=5
30
- # ) # https://huggingface.co/blog/constrained-beam-search
31
- # temperature = st.sidebar.slider(
32
- # "Temperature", value=0.9, min_value=0.0, max_value=1.0, step=0.05
33
- # )
34
- # top_k = st.sidebar.slider("Top-k", min_value=0, max_value=5, value=0)
35
- # top_p = st.sidebar.slider("Top-p", min_value=0.0, max_value=1.0, step=0.05, value=1.0)
36
-
37
 
38
  def infer_sentence(model, src_text, tokenizer=tokenizer):
39
  encoded_prompt = tokenizer.encode(
@@ -79,7 +70,6 @@ if st.button("문장 변환") or src_text == default_value:
79
  st.warning("Please **enter text** for translation")
80
 
81
  else:
82
- st.success("Translating...")
83
  english_translation = infer_sentence(
84
  model=ko2en_model, src_text=src_text, tokenizer=tokenizer
85
  )
@@ -103,7 +93,7 @@ if st.button("문장 변환") or src_text == default_value:
103
  clean_up_tokenization_spaces=True,
104
  skip_special_tokens=True,
105
  )
106
- st.success(f"{src_text} -> {english_translation} -> {korean_translation}")
107
  else:
108
  pass
109
 
 
25
  )
26
  print(src_text)
27
 
 
 
 
 
 
 
 
 
 
28
 
29
  def infer_sentence(model, src_text, tokenizer=tokenizer):
30
  encoded_prompt = tokenizer.encode(
 
70
  st.warning("Please **enter text** for translation")
71
 
72
  else:
 
73
  english_translation = infer_sentence(
74
  model=ko2en_model, src_text=src_text, tokenizer=tokenizer
75
  )
 
93
  clean_up_tokenization_spaces=True,
94
  skip_special_tokens=True,
95
  )
96
+ print(f"{src_text} -> {english_translation} -> {korean_translation}")
97
  else:
98
  pass
99