| import streamlit as st | |
| from transformers import pipeline | |
| # Load your model | |
| model_name = "your-username/your-model-name" | |
| qa_pipeline = pipeline("text-generation", model=model_name) | |
| # Streamlit app | |
| st.title("Investment Banking Assistant") | |
| user_input = st.text_input("Ask a question:") | |
| if user_input: | |
| response = qa_pipeline(user_input, max_length=100) | |
| st.write(response[0]["generated_text"]) |