gpt2-deployment / app.py
tchans123's picture
Update app.py
b7f1129 verified
raw
history blame
1.56 kB
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)