File size: 681 Bytes
d050156 a7a26ba d050156 c758640 d050156 | 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 28 29 | import streamlit as st
from langchain_community.llms import HuggingFaceHub
# from langchain.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) |