| | import streamlit as st |
| |
|
| | from langchain_community.llms import HuggingFaceHub |
| | |
| |
|
| | def load_answer(question): |
| | llm = HuggingFaceHub(repo_id = "google/flan-t5-large") |
| | ans = llm.invoke(question) |
| | return ans |
| |
|
| | st.set_page_config(page_title="LangChain Demo", page_icon=":robot:") |
| | st.header("LangChain Demo") |
| |
|
| | def get_text(): |
| | input_text = st.text_input("You: ", key="input") |
| | return input_text |
| |
|
| | user_input=get_text() |
| |
|
| | if not user_input: |
| | st.warning("Please enter a question before generating") |
| | else: |
| | response=load_answer(user_input) |
| |
|
| | submit=st.button('Generate') |
| |
|
| | if submit: |
| | st.subheader('Answer:') |
| | st.write(response) |
