Spaces:
Sleeping
Sleeping
import gradio as gr | |
import openai | |
from huggingface_hub import HfApi | |
import os | |
# Initialize Hugging Face API to fetch secrets | |
api = HfApi() | |
# Fetch the OpenAI API key from Hugging Face secrets | |
openai.api_key = api.get_repo_secret( | |
repo_id="tchans123/gpt2-deployment", # Replace with your Hugging Face repository ID | |
secret_name="OPENAI_API_KEY" # Name of the secret in Hugging Face | |
) | |
def generate_text(prompt, max_length=100, temperature=1.0): | |
try: | |
# Generate text using OpenAI's GPT-3 (text-davinci-003) | |
response = openai.Completion.create( | |
engine="text-davinci-003", # Specify the engine | |
prompt=prompt, | |
max_tokens=max_length, | |
temperature=temperature, | |
n=1, | |
stop=None | |
) | |
return response.choices[0].text.strip() | |
except openai.error.OpenAIError as e: | |
return f"An OpenAI error occurred: {str(e)}" | |
except Exception as e: | |
return f"An unexpected error occurred: {str(e)}" | |
# Create a Gradio interface | |
iface = gr.Interface( | |
fn=generate_text, | |
inputs=[ | |
gr.Textbox(lines=3, placeholder="Enter your prompt here...", label="Prompt"), | |
gr.Slider(minimum=50, maximum=500, value=100, label="Max Length"), | |
gr.Slider(minimum=0.1, maximum=1.5, value=1.0, label="Temperature"), | |
], | |
outputs=gr.Textbox(label="Generated Text"), | |
title="GPT Text Generator", | |
description="Enter a prompt and generate text using OpenAI's GPT (text-davinci-003)." | |
) | |
# Launch the interface | |
iface.launch(share=True) | |