gpt-playground / app.py
keisuke-tada's picture
Update app.py
280a867
raw
history blame
642 Bytes
import openai
import streamlit as st
openai.api_key = os.getenv("OPENAI_API_KEY")
temperature = 0
input = st.text_input("Input")
if st.button("Submit", type="primary"):
st.markdown("----")
res_box = st.empty()
content = []
for chunk in openai.Completion.create(
model="gpt-3.5-turbo",
temperature=temperature,
messages=[{"role": "user", "content": input}],
stream=True,
):
content.append(chunk["choices"][0].get("delta", {}).get("content"))
result = "".join(content).strip()
result = result.replace("\n", "")
res_box.markdown(result)
st.markdown("----")