Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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 = "
|
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)
|