gokceuludogan commited on
Commit
2ecade9
·
verified ·
1 Parent(s): 98793d5

Update apps/home.py

Browse files
Files changed (1) hide show
  1. apps/home.py +1 -1
apps/home.py CHANGED
@@ -99,7 +99,7 @@ def write():
99
  url = "https://api-inference.huggingface.co/models/boun-tabi-LMG/TURNA"
100
  params = {"length_penalty": length_penalty, "no_repeat_ngram_size": no_repeat_ngram_size, "max_new_tokens": max_new_tokens,
101
  "do_sample":do_sample, "num_beams":num_beams, "repetition_penalty":repetition_penalty,
102
- "top_p":top_p, "top_k":top_k, "temperature":temp}
103
  if st.button("Generate"):
104
  with st.spinner('Generating...'):
105
  output = query(input_text, url, params)
 
99
  url = "https://api-inference.huggingface.co/models/boun-tabi-LMG/TURNA"
100
  params = {"length_penalty": length_penalty, "no_repeat_ngram_size": no_repeat_ngram_size, "max_new_tokens": max_new_tokens,
101
  "do_sample":do_sample, "num_beams":num_beams, "repetition_penalty":repetition_penalty,
102
+ "top_p":top_p, "top_k":top_k, "temperature":temp, "early_stopping": True}
103
  if st.button("Generate"):
104
  with st.spinner('Generating...'):
105
  output = query(input_text, url, params)