|
|
|
|
|
|
|
import transformers |
|
import gradio as gr |
|
import torch |
|
|
|
from transformers import AutoTokenizer, AutoModelForSeq2SeqLM |
|
|
|
tokenizer = AutoTokenizer.from_pretrained("humarin/chatgpt_paraphraser_on_T5_base") |
|
model = AutoModelForSeq2SeqLM.from_pretrained("humarin/chatgpt_paraphraser_on_T5_base") |
|
|
|
def paraphrase( |
|
Content_to_Rephrase, |
|
num_beams: 3, |
|
num_return_sequences=3, |
|
): |
|
input_ids = tokenizer( |
|
f'paraphrase: {Content_to_Rephrase}', |
|
return_tensors="pt", padding="longest", |
|
|
|
).input_ids |
|
|
|
outputs = model.generate(input_ids, num_beams=num_beams, num_return_sequences=num_return_sequences) |
|
|
|
res = tokenizer.batch_decode(outputs, skip_special_tokens=True) |
|
res1 = res [0] |
|
res2 = res [1] |
|
res3 = res [2] |
|
|
|
|
|
return res1, res2, res3 |
|
|
|
output1 = gr.Textbox(label="Rephrased: Option 1") |
|
output2 = gr.Textbox(label="Rephrased: Option 2") |
|
output3 = gr.Textbox(label="Rephrased: Option 3") |
|
|
|
input = gr.Textbox(lines=10) |
|
|
|
iface = gr.Interface(fn=paraphrase, |
|
inputs=[input], |
|
outputs=[output1, output2, output3], |
|
title="Free AI Sentence Rephraser", |
|
description="<ul><li>Paste text in the input box and press 'Submit'.</li><li>Max length: ~35 words (larger content is summarized)</li><li>The rephrased sentences *may not* be better than the original input.</li><li>Model 'humarin' pre-trained by ChatGPT. Temp = 0.7</li></ul>", |
|
examples=[ |
|
["With the humble is wisdom."], |
|
["Hatred stirs up strife."], |
|
["The way of a fool is right in his own eyes."], |
|
["Righteousness leads to life."], |
|
], |
|
cache_examples=True, |
|
) |
|
|
|
iface.launch() |
