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"])