Update

app.py

@@ -10,22 +10,24 @@ model = T5ForConditionalGeneration.from_pretrained(model_path)
 tokenizer = AutoTokenizer.from_pretrained(model_path)
 pipe = pipeline(task='text2text-generation', model=model, tokenizer=tokenizer)
 
-def paraphrase(text, min_len, max_len=128, n_beams=5):
-  for j in range(3):
-    out = pipe(text, encoder_no_repeat_ngram_size=4, do_sample=True, num_beams=n_beams, max_length=max_len)[0]['generated_text']
+def paraphrase(text, num_outputs, n_beams, min_len, max_len):
+#   for j in range(3):
+    out = pipe(text, num_outputs=num_outputs, encoder_no_repeat_ngram_size=4, do_sample=True, num_beams=n_beams, max_length=max_len)[0]['generated_text']
     return out
 
 text = st.text_area('متن محاوره‌ای خود را در این کادر وارد نمائید:', "من با دوستام میرم بازی", height=50)
+num_outputs = st.sidebar.slider('تعداد خروجی', min_value=1, max_value=10, value=3, step=1)
 num_beams = st.sidebar.slider('طول Beam', min_value=1, max_value=10, value=5, step=1)
-min_length = st.sidebar.slider('حداقل طول جمله', min_value=10, max_value=100, value=50, step=10)
-max_length = st.sidebar.slider('حداکثر طول طول جمله', min_value=30, max_value=700, value=100, step=10)
+min_length = st.sidebar.slider('حداقل طول جمله', min_value=10, max_value=100, value=30, step=5)
+max_length = st.sidebar.slider('حداکثر طول طول جمله', min_value=30, max_value=128, value=100, step=5)
 button = st.button("تبدیل متن")
 
 with st.spinner("در حال تبدیل متن..."):
     if button and text:
-        out = paraphrase(text, num_beams, min_length, max_length)
+        out = paraphrase(text,num_outputs, num_beams, min_length, max_length)
         st.markdown("**متن خروجی:**")
-        st.write(out)
+        for i in out:
+            st.write(out)
 
 # x = st.slider('Select a value')
 # st.write(x, 'squared is', x * x)