erfan226 commited on
Commit
2e8ee0a
1 Parent(s): 7230fc3
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -12,7 +12,7 @@ pipe = pipeline(task='text2text-generation', model=model, tokenizer=tokenizer)
12
 
13
  def paraphrase(text, num_outputs, n_beams, min_len, max_len):
14
  # for j in range(3):
15
- out = pipe(text, num_return_sequences=num_outputs, encoder_no_repeat_ngram_size=4, do_sample=True, num_beams=n_beams, max_length=max_len)[0]['generated_text']
16
  return out
17
 
18
  text = st.text_area('متن محاوره‌ای خود را در این کادر وارد نمائید:', "من با دوستام میرم بازی", height=50)
@@ -26,8 +26,8 @@ with st.spinner("در حال تبدیل متن..."):
26
  if button and text:
27
  out = paraphrase(text,num_outputs, num_beams, min_length, max_length)
28
  st.markdown("**متن خروجی:**")
29
- for i in out:
30
- st.write(out)
31
 
32
  # x = st.slider('Select a value')
33
  # st.write(x, 'squared is', x * x)
 
12
 
13
  def paraphrase(text, num_outputs, n_beams, min_len, max_len):
14
  # for j in range(3):
15
+ out = pipe(text, num_return_sequences=num_outputs, encoder_no_repeat_ngram_size=4, do_sample=True, num_beams=n_beams, max_length=max_len)
16
  return out
17
 
18
  text = st.text_area('متن محاوره‌ای خود را در این کادر وارد نمائید:', "من با دوستام میرم بازی", height=50)
 
26
  if button and text:
27
  out = paraphrase(text,num_outputs, num_beams, min_length, max_length)
28
  st.markdown("**متن خروجی:**")
29
+ for i, txt in enumerate(out):
30
+ st.write(out[i]['generated_text'])
31
 
32
  # x = st.slider('Select a value')
33
  # st.write(x, 'squared is', x * x)