Dagfinn1962's picture
Update app.py
8caf31f
raw
history blame
1.14 kB
import os
import openai
import gradio as gr
openai.ChatCompletion.create(
model="gpt-3.5-turbo0301",
# Set OpenAI API key from environment variable
openai.api_key = os.environ.get("OPENAI_API_KEY")
# Define GPT-3.5 Turbo model
model = "gpt-3.5-turbo-0301"
# Define function for generating responses
def generate_response(prompt):
try:
response = openai.Completion.create(
engine=model,
prompt=prompt,
max_tokens=1024,
n=1,
stop=None,
temperature=0.7,
)
return response.choices[0].text.strip()
except Exception as e:
return f"Error: {e}"
# Define input and output interfaces
input_text = gr.inputs.Textbox(label="Enter your message:")
output_text = gr.outputs.Textbox(label="AI response:")
# Define Gradio app
gradio_app = gr.Interface(
fn=generate_response,
inputs=input_text,
outputs=output_text,
title="GPT-3.5 Turbo Chatbot",
description="Enter a message and get a response from an AI chatbot powered by GPT-3.5 Turbo.",
theme="compact",
)
# Launch Gradio app
gradio_app.launch(share=True)