Spaces:
Sleeping
Sleeping
File size: 668 Bytes
d35c879 b307974 d35c879 b307974 d35c879 b307974 d35c879 b307974 d35c879 b307974 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 |
import os
import openai
import gradio as gr
# Set OpenAI API Key
OPENAI_API_KEY = os.getenv('OPENAI_API_KEY')
openai.api_key = OPENAI_API_KEY
def get_text_response(user_message, history):
# Call OpenAI GPT model
response = openai.ChatCompletion.create(
model="gpt-3.5-turbo",
messages=[
{"role": "system", "content": "You are a helpful assistant."},
{"role": "user", "content": user_message},
]
)
return response['choices'][0]['message']['content']
# Create a Gradio chat interface
demo = gr.Interface(fn=get_text_response, inputs="text", outputs="text")
if __name__ == "__main__":
demo.launch()
|