Article / app.py
mano96's picture
Update app.py
4f5bdfd
import gradio as gr
import openai
import re
# Authenticate with OpenAI API
openai.api_key = "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"
# Set the model and prompt for GPT-2
model_engine = "text-davinci-002"
model_prompt = "Rewrite this text:"
# Define the function to generate the paraphrased output
def paraphrase(text):
# Format the input text for the model prompt
formatted_text = f"{model_prompt} {text}"
# Generate the paraphrased output
response = openai.Completion.create(
engine=model_engine,
prompt=formatted_text,
max_tokens=1024,
n=1,
stop=None,
temperature=0.7,
)
# Extract the paraphrased text from the response
paraphrased_text = response.choices[0].text
# Clean up the paraphrased text
paraphrased_text = re.sub(model_prompt, "", paraphrased_text)
paraphrased_text = paraphrased_text.strip()
return paraphrased_text
# Create the user interface
input_text = gr.inputs.Textbox(lines=20, label='Input Text')
output_text = gr.outputs.Textbox(label='Output Text')
app = gr.Interface(paraphrase, inputs=input_text,css="""span.svelte-1l2rj76{color: #591fc9;font-size: 18px;
font-weight: 600;}.secondary.svelte-1ma3u5b{background: #591fc9; color: #fff;}.secondary.svelte-1ma3u5b:hover{background:#8a59e8;color:#000;}
.svelte-2xzfnp textarea {border: 1px solid #591fc9}.primary.svelte-1ma3u5b{background: #f8d605;color: #000;}.primary.svelte-1ma3u5b:hover{background: #ffe751;color: #591fc9;} """, outputs=output_text, title='GPT-2 Article Rewriter', description='Generate a paraphrased output while preserving the meaning of the input text using OpenAI GPT-2 model.')
# Launch the user interface
app.launch(inline=False)