Spaces:
Runtime error
Runtime error
Fix
Browse files
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,
|
16 |
return out
|
17 |
|
18 |
text = st.text_area('متن محاورهای خود را در این کادر وارد نمائید:', "من با دوستام میرم بازی", height=50)
|
|
|
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)
|