Spaces:
Runtime error
Runtime error
import gradio as gr | |
import openai | |
import time | |
# Authenticate with OpenAI API | |
openai.api_key = "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" | |
# Define GPT-2 model | |
model_engine = "text-davinci-002" | |
prompt = "Write a short paragraph about a topic of your choice:" | |
def generate_text(prompt): | |
# Use GPT-2 to generate text based on user input | |
response = openai.Completion.create( | |
engine=model_engine, | |
prompt=prompt, | |
max_tokens=1024, | |
n=1, | |
stop=None, | |
temperature=0.5, | |
) | |
generated_text = response.choices[0].text.strip() | |
return generated_text | |
def generate_text_slowly(prompt): | |
# Generate text character by character, with a delay between each character | |
generated_text = "" | |
response = openai.Completion.create( | |
engine=model_engine, | |
prompt=prompt, | |
max_tokens=1024, | |
n=1, | |
stop=None, | |
temperature=0.5, | |
) | |
generated_chars = response.choices[0].text.strip() | |
for char in generated_chars: | |
generated_text += char | |
time.sleep(0.05) # Add a small delay between characters to simulate typing | |
return generated_text | |
# Create Gradio interface | |
interface = gr.Interface( | |
generate_text_slowly, | |
inputs=gr.inputs.Textbox(prompt), | |
outputs=gr.outputs.Textbox(), | |
title="GPT-2 Text Generation", | |
description="Enter a short prompt and watch as GPT-2 generates text based on your input.", | |
) | |
# Launch the interface | |
interface.launch() | |