mildo1x2 commited on
Commit
d4d7a43
·
verified ·
1 Parent(s): ca7e81e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -7,7 +7,7 @@ tokenizer = T5Tokenizer.from_pretrained(model_name, use_fast=False)
7
  model = MT5ForConditionalGeneration.from_pretrained(model_name)
8
 
9
  def humanize(text):
10
- prompt = "Parafrázuj tento text prirodzene a ľudsky: " + text
11
  inputs = tokenizer(prompt, return_tensors="pt", truncation=True, padding=True)
12
  outputs = model.generate(**inputs, max_length=512, num_beams=4)
13
  return tokenizer.decode(outputs[0], skip_special_tokens=True)
 
7
  model = MT5ForConditionalGeneration.from_pretrained(model_name)
8
 
9
  def humanize(text):
10
+ prompt = "paraphrase: " + text
11
  inputs = tokenizer(prompt, return_tensors="pt", truncation=True, padding=True)
12
  outputs = model.generate(**inputs, max_length=512, num_beams=4)
13
  return tokenizer.decode(outputs[0], skip_special_tokens=True)