| import streamlit as st | |
| from transformers import pipeline | |
| model_id = "microsoft/Phi-3-mini-4k-instruct" | |
| pipe = pipeline("text-generation", model=model_id) | |
| def load_answer(question): | |
| return pipe(question, max_length=200)[0]['generated_text'] | |
| st.set_page_config(page_title='LangChain Demo', page_icon=':robot:') | |
| st.header("LangChain Demo") | |
| def get_text(): | |
| return st.text_input("Question: ", key="input") | |
| user_input = get_text() | |
| submit = st.button("Ask") | |
| if submit: | |
| output = load_answer(user_input) | |
| st.write(output) |