Spaces:
Sleeping
Sleeping
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) |