GradioTest / app.py
dlflannery's picture
Update app.py
ddd0404 verified
raw
history blame
501 Bytes
import os
import gradio as gr
# import openai
from openai import OpenAI
#from python-dotenv import load_dotenv
#load_dotenv()
key = os.getenv('OPENAI_API_KEY')
client = OpenAI(api_key = key)
def chat(prompt):
completion = client.chat.completions.create(model="gpt-4o-mini",
messages=[{"role":"user", "content":prompt}])
return completion.choices[0].message.content
demo = gr.Interface(fn=chat, inputs="text", outputs="text")
demo.launch()