File size: 619 Bytes
d055f50 |
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 26 27 |
import streamlit as st
from langchain.llms import OpenAI
#App UI starts here
st.set_page_config(page_title='LangChain Demo', page_icon=":robot:")
st.header('LangChain Demo')
#Get user input
def get_text():
input_text = st.text_input("You: ", key="input")
return input_text
#Generate response
def generate_response(user_input):
llm = OpenAI(model_name="text-davinci-003", temperature=0)
response = llm(user_input)
return response
user_input = get_text()
submit = st.button("Generate")
if submit:
st.subheader("Answer: ")
response = generate_response(user_input)
st.write(response) |