BigSalmon commited on
Commit
21a1f5b
·
1 Parent(s): 8e3d79e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
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,