schakrav/LLMSTest

1import streamlit as st2 3 4from langchain.llms import OpenAI5 6#Function to return the response7def load_answer(question):8    llm = OpenAI(model_name="text-davinci-003",temperature=0)9    answer=llm(question)10    return answer11 12 13#App UI starts here14st.set_page_config(page_title="LLM Demo", page_icon=":shark:",initial_sidebar_state="expanded")15st.header("LLM Demo")16 17#Gets the user input18def get_text():19    input_text = st.text_input("You: ", key="input")20    return input_text21 22 23user_input=get_text()24response = load_answer(user_input)25 26submit = st.button('Generate')  27 28#If generate button is clicked29if submit:30 31    st.subheader("Answer:")32 33    st.write(response)