File size: 399 Bytes
9bd058b |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 |
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"]) |