r1208 commited on
Commit
68c89fe
·
verified ·
1 Parent(s): 46e037c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -83,7 +83,7 @@ def main():
83
 
84
  if st.button("Translate"):
85
  if input_text:
86
- translation = translate(input_text, model, do_sample = do_sample, max_new_tokens = max_new_tokens, temperature = temperature, top_k = top_k, top_p = top_p, bad_words_ids = bad_words_ids)
87
  st.text_area("Translated Text", value=translation, height=300)
88
  else:
89
  st.error("Please enter some text to translate.")
 
83
 
84
  if st.button("Translate"):
85
  if input_text:
86
+ translation = translate(input_text, tokenizer, model, do_sample = do_sample, max_new_tokens = max_new_tokens, temperature = temperature, top_k = top_k, top_p = top_p, bad_words_ids = bad_words_ids)
87
  st.text_area("Translated Text", value=translation, height=300)
88
  else:
89
  st.error("Please enter some text to translate.")