| import streamlit as st | |
| # from langchain.llms import OpenAI | |
| from langchain.llms import HuggingFace | |
| ## Function to return the response | |
| def load_answer(question): | |
| # llm = OpenAI(model_name="gpt-3.5-turbo") | |
| llm = HuggingFace(model_name="google/flan-t5-large") | |
| answer = llm(question) | |
| return answer | |
| ## App UI starts here | |
| st.set_page_config(page_title="LangChain Demo", page_icon="🦜") | |
| st.header("🦜️🔗 LangChain Demo") | |
| ## Gets the user input | |
| def get_text(): | |
| input_text = st.text_input("You: ", key="input") | |
| return input_text | |
| user_input = get_text() | |
| response = load_answer(user_input) | |
| submit = st.button("Generate") | |
| ## If generate button is clicked | |
| if submit: | |
| st.subheader("Answer:") | |
| st.write(response) |