SimpleApp / app.py
gautamraj8044's picture
Create app.py
1079beb verified
raw
history blame
547 Bytes
import streamlit as st
import langchain
from langchain.llms import HuggingFaceHub
def load_answer(question):
llm = HuggingFaceHub(repo_id="google/flan-t5-large")
answer = llm(question)
return answer
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()
response = load_answer(user_input)
submit = st.button("Generate")
if submit:
st.subheader("Answer:")
st.write(response)