Update app.py
Browse files
app.py
CHANGED
@@ -7,6 +7,7 @@ st.title("Paraphrase")
|
|
7 |
def get_model():
|
8 |
tokenizer = AutoTokenizer.from_pretrained("facebook/bart-large-cnn")
|
9 |
model = AutoModelForSeq2SeqLM.from_pretrained("facebook/bart-large-cnn")
|
|
|
10 |
return model, tokenizer
|
11 |
|
12 |
model, tokenizer = get_model()
|
@@ -25,6 +26,7 @@ def translate_to_english(model, tokenizer, text):
|
|
25 |
input_ids=input_ids, attention_mask=attention_masks,
|
26 |
do_sample=True,
|
27 |
max_length=256,
|
|
|
28 |
temperature = temp,
|
29 |
top_k=120,
|
30 |
top_p=0.98,
|
|
|
7 |
def get_model():
|
8 |
tokenizer = AutoTokenizer.from_pretrained("facebook/bart-large-cnn")
|
9 |
model = AutoModelForSeq2SeqLM.from_pretrained("facebook/bart-large-cnn")
|
10 |
+
|
11 |
return model, tokenizer
|
12 |
|
13 |
model, tokenizer = get_model()
|
|
|
26 |
input_ids=input_ids, attention_mask=attention_masks,
|
27 |
do_sample=True,
|
28 |
max_length=256,
|
29 |
+
min_length = 100,
|
30 |
temperature = temp,
|
31 |
top_k=120,
|
32 |
top_p=0.98,
|