File size: 510 Bytes
4ec53ca
2bedd68
4ec53ca
 
 
b20a07f
4ec53ca
b20a07f
2bedd68
 
4ec53ca
 
16ed246
4ec53ca
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19

import os
import gradio as gr
# import openai
from openai import OpenAI
from 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()