Update app.py

app.py

@@ -16,7 +16,7 @@ def main():
         return tokens_list
 
 
-    def translate(text):
+    def translate(text, tokenizer, model, bad_words_ids)):
         # Prepare the prompt
         messages = f"Translate from Korean to English: {text}"
         input_ids = tokenizer.apply_chat_template(messages, tokenize=True, add_generation_prompt=True, return_tensors="pt")
@@ -68,7 +68,7 @@ def main():
 
     bad_words_ids = get_tokens_as_list( word_list=["\n", "\n\n", "\ ", " \ ", "\\", "'\n'"] )
 
-    max_new_tokens = st.sidebar.slider("Max Length", value=100, min_value=10, max_value=1000)
+    max_new_tokens = st.sidebar.slider("Max Length", value=400, min_value=10, max_value=1000)
     temperature = st.sidebar.slider("Temperature", value=0.3, min_value=0.0, max_value=1.0, step=0.05)
     top_k = st.sidebar.slider("Top-k", min_value=0, max_value=50, value=0)
     top_p = st.sidebar.slider("Top-p", min_value=0.75, max_value=1.0, step=0.05, value=0.9)