gokceuludogan commited on
Commit
04ab1d5
1 Parent(s): ca5d5f0

Fix summarization

Browse files
Files changed (1) hide show
  1. apps/summarization.py +1 -1
apps/summarization.py CHANGED
@@ -72,7 +72,7 @@ def write():
72
  input_text = st.text_area(label='Enter a text: ', height=200,
73
  value="Kalp krizi geçirenlerin yaklaşık üçte birinin kısa bir süre önce grip atlattığı düşünülüyor. Peki grip virüsü ne yapıyor da kalp krizine yol açıyor? Karpuz şöyle açıkladı: Grip virüsü kanın yapışkanlığını veya pıhtılaşmasını artırıyor.")
74
  url = ("https://api-inference.huggingface.co/models/boun-tabi-LMG/" + model_name.lower())
75
- params = {"length_penalty": length_penalty, "no_repeat_ngram_size": no_repeat_ngram_size, "max_new_tokens": max_new_tokens }
76
  if st.button("Generate"):
77
  with st.spinner('Generating...'):
78
  output = query(input_text, url, params)
 
72
  input_text = st.text_area(label='Enter a text: ', height=200,
73
  value="Kalp krizi geçirenlerin yaklaşık üçte birinin kısa bir süre önce grip atlattığı düşünülüyor. Peki grip virüsü ne yapıyor da kalp krizine yol açıyor? Karpuz şöyle açıkladı: Grip virüsü kanın yapışkanlığını veya pıhtılaşmasını artırıyor.")
74
  url = ("https://api-inference.huggingface.co/models/boun-tabi-LMG/" + model_name.lower())
75
+ params = {"length_penalty": length_penalty, "no_repeat_ngram_size": no_repeat_ngram_size, "max_new_tokens": max_new_tokens, "decoder_start_token_id": 0, "eos_token_id": 1, "pad_token_id": 0 }
76
  if st.button("Generate"):
77
  with st.spinner('Generating...'):
78
  output = query(input_text, url, params)