streaming / app.py
joseluhf11's picture
Update app.py
303dfad verified
import gradio as gr
from openai import OpenAI
def streaming(api_key, template, prompt):
openai_client = OpenAI(api_key=api_key)
messages = [
{"role": "system", "content": template},
{"role": "user", "content": prompt}
]
response = openai_client.chat.completions.create(
model="gpt-4-turbo-preview",
messages=messages,
temperature = 0,
stream = True
)
full_response = ""
for chunk in response:
if chunk.choices[0].delta.content is not None:
full_response += chunk.choices[0].delta.content
yield full_response
inputs = [
gr.Textbox(label="API KEY"),
gr.Textbox(label = 'Template'),
gr.Textbox(label="Prompt")
]
output = gr.Textbox()
app = gr.Interface(streaming, inputs, output)
app.launch()